models cannot load - Unknown flags in XLA_FLAGS

[adrienchaton]

Hi,

I am trying to run Proteinfer locally, requirements seem to install without issues and imports go through when running proteinfer.py.
But when it tries to load the tensorflow models, it crashes because of

tensorflow/compiler/xla/parse_flags_from_env.cc:221] Unknown flags in XLA_FLAGS: --xla_gpu_enable_triton_gemm=false
Perhaps you meant to specify these on the TF_XLA_FLAGS envvar?

I tried to update to different versions of tensorflow packages but I could not get anything to fix this issue...
Any hints please?