killabee777 / examples /text_to_image /requirements_flax.txt
danielpikl's picture
Add stable diffusion weights
af7c068
diffusers>==0.5.1
transformers>=4.21.0
flax
optax
torch
torchvision
ftfy
tensorboard
modelcards