flux-dev-flax / requirements.txt
lnyan's picture
Update
d4607d7
raw
history blame
175 Bytes
jax[cuda12]
flax==0.9.0
flash_attn_jax
torch
torchvision
opencv-python-headless
einops
huggingface_hub
transformers
tokenizers
sentencepiece
fire
invisible-watermark
ml-dtypes