TF2XLA Conversion Failed

,

Hi, I get the following error message when I try this tutorial from huggingface. Translation - Hugging Face NLP Course

tf2xla conversion failed while converting __inference_generate_with_xla_1354863[_XlaMustCompile=true,config_proto=6001324581131673121,executor_type=11160318154034397263]. Run with TF_DUMP_GRAPH_PREFIX=/path/to/dump/dir and --vmodule=xla_compiler=2 to obtain a dump of the compiled functions. [Op:__inference_generate_with_xla_1354863]