torch==2.0.1
accelerate==0.21.0
bitsandbytes==0.41.0
deepspeed==0.9.5
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
wavedrom==2.0.3.post3
Pygments==2.16.1
albumentations==0.4.3
opencv-python==4.1.2.30
pudb==2019.2
invisible-watermark
imageio==2.9.0
imageio-ffmpeg==0.4.2
pytorch-lightning==1.4.2
omegaconf==2.1.1
test-tube>=0.7.5
streamlit>=0.73.1
torch-fidelity==0.3.0
torchmetrics==0.6.0
kornia==0.6