requirements.txt

# PyTorch with CUDA support (for GPU acceleration)
# Install from PyTorch official channel for CUDA 11.8 or 12.1
torch>=2.0.0
torchvision>=0.15.0
torchaudio>=2.0.0

# Diffusion models and related packages
diffusers 
transformers 
accelerate 
scipy 
safetensors 

# Optional for better performance
xformers

# Gradio for web interface
gradio