ModuleNotFoundError: No module named 'flaxformer'

✨Check the date of this article before reading!✨

No module named ‘flaxformer’

Recently, Google used more and more models built on JAX and Flax. When I try to run Vision Transformer pretrained models created by Google, I got the following error.

ModuleNotFoundError: No module named 'flaxformer'

But I already installed the requirements.txt, I was confused about the error. And the flaxformer can not be installed by using pip install [package name] since it did not release on PyPI.

Then I checked the repositories of flax and flaxformer:
https://github.com/google/flax
https://github.com/google/flaxformer

Fix

The module flaxformer was not released, so it can be installed by:

pip install git+https://github.com/google/flaxformer

Solved.🎉