TF5 XLA error

[SoyGema]

After training and during inference this error was found

1092, in xla_dynamic_slice_eager_fallback
    _result = _execute.execute(b"XlaDynamicSlice", 1, inputs=_inputs_flat,
tensorflow.python.framework.errors_impl.UnimplementedError: Exception encountered when calling layer "SelfAttention" (type TFT5Attention).

Could not find compiler for platform METAL: NOT_FOUND: could not find registered compiler for platform METAL -- check target linkage [Op:XlaDynamicSlice]

Follow up

• Read forum
• Saw that argument use_xla_generation in script was set to False.
• Visit argument definition

[SoyGema]

Changing the argument to True seems to give same error in some sense

File "/Users/gema/Documents/transformers/src/transformers/keras_callbacks.py", line 223, in on_epoch_end
    predictions = self.generation_function(generation_inputs, attention_mask=attention_mask)
tensorflow.python.framework.errors_impl.UnimplementedError: Could not find compiler for platform METAL: NOT_FOUND: could not find registered compiler for platform METAL -- check target linkage [Op:__inference_generation_function_28937]

Try : changing tf-metal install