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.
user Cahya on discord helped me on this, turns out FlaxVisionEncoderDecoderModel is not on 4.12.5, so simply install it from the repo in order to have it!