torch==2.0.1
bitsandbytes==0.41.0
diffusers
einops-exts==0.0.4
einops==0.6.1
gradio==3.35.2
gradio_client==0.2.9
httpx==0.24.0
markdown2==2.4.10
numpy==1.26.0
peft==0.4.0
scikit-learn==1.2.2
sentencepiece==0.1.99
shortuuid==1.0.11
timm==0.6.13
tokenizers==0.13.3
torchvision==0.15.2
transformers==4.31.0
wandb==0.15.12
wavedrom==2.0.3.post3
ftfy>=6.0
future==0.18.0
hydra-core>=1.1
loguru>=0.6
nltk>=3.6
tensorboard==2.14.0
tensorflow-datasets>=4.9.0
torchmetrics>=1.0
tqdm>=4.50.0
ujson>=5.0
webdataset>=0.2
wordsegment>=1.3