So I tried doing from transformers import FlaxVisionEncoderDecoderModel but it throws me the following exception ImportError: cannot import name 'FlaxVisionEncoderDecoderModel' from 'transformers' (/.local/lib/python3.8/site-packages/transformers/__init__.py)
I have Flax installed, (with jax as well) I upgraded Transformers to the latest version.
As a reference: I can run GPT-J with transformers on this TPUVM just fine, so I don’t know why it doesnt like this flax model-thing?
I’m totally lost, help needed.