torch==2.0.1
torchvision
torchaudio
xformers>=0.0.20
omegaconf
hydra-core
pyrootutils
transformers==4.30.2
flask
gradio==3.28.3
timm
diffusers==0.20.2
accelerate
scipy
bitsandbytes
einops
sentencepiece