requirements.txt

triton
pydantic==1.10.9
accelerate==0.27.2
imageio==2.25.1
numpy==1.26.4
omegaconf==2.3.0
tokenizers==0.13.2
transformers==4.26.1
diffusers==0.28.2
sentencepiece==0.1.99
requests