requirements.txt

# Core ML libraries
torch>=2.1.0
torchvision>=0.16.0

# Diffusers & Transformers
diffusers>=0.23.1
transformers>=4.35.0
sentencepiece>=0.1.99

# Hugging Face pipelines
torchmetrics>=1.1.0
accelerate>=0.24.0

# Image quality & metrics
lpips>=0.1.4
clip @ git+https://github.com/openai/CLIP.git

# NLP evaluation
bert-score>=0.3.12

# Image processing
Pillow>=10.0.0
torchvision>=0.16.0

# Gradio
gradio>=3.50.0

# Optional GPU dependencies
scipy>=1.11.0
numpy>=1.26.0