requirements.txt

fire
triton
pydantic
accelerate==0.27.2
deepspeed==0.8.1
huggingface-hub==0.20.2
imageio==2.25.1
numpy==1.21.6
omegaconf==2.3.0
pandas==1.3.5
Pillow==9.4.0
tokenizers==0.13.2
torch==1.13.1
torchvision==0.14.1
transformers==4.26.1
xformers==0.0.16
safetensors==0.3.3
diffusers==0.28.2
sentencepiece==0.1.99
gradio==3.38.0
matplotlib-inline
ipython