requirements.txt

transformers
accelerate
python-slugify
python-dotenv
einops
timm
autotrain-advanced
torch
safetensors
diffusers
flash-attn
ai-toolkit
gradio
huggingface_hub

omegaconf
einops
xformers