torch
torchsde
torchvision
torchaudio
einops
transformers>=4.28.1
tokenizers>=0.13.3
sentencepiece
safetensors>=0.4.2
pyyaml
Pillow
scipy
tqdm
psutil
accelerate
huggingface_hub
xformers>=0.0.27
