# For flash-attn
wheel
packaging
ninja
einops
# flash-attn==2.5.8

scikit-image
diffusers  # used for stable diffusion tokenizers

# datasets
opencv-python
tensorflow_datasets
tensorflow
gcsfs

# official MaskGIT
flax