requirements.txt

# CORE GPU LIBRARIES (PyTorch - GPU ONLY)
# IMPORTANT: Install manually using:
# pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

torch==2.5.1
torchvision==0.20.1
torchaudio==2.5.1


# HUGGINGFACE DIFFUSION ECOSYSTEM
diffusers==0.35.2
transformers==4.57.3    
huggingface_hub==0.36.0
accelerate==0.33.0
safetensors==0.7.0
peft==0.17.1


# UI FRAMEWORK
gradio==6.0.2


# IMAGE PROCESSING & UTILITIES
opencv-python==4.9.0.80
Pillow==10.2.0
numpy==1.26.4
loguru==0.7.2
tqdm==4.66.2
python-dotenv==1.0.1


# UPSCALING / SUPER-RESOLUTION
realesrgan-ncnn-py==2.0.0


# DEVELOPMENT TOOLS
black==24.3.0
ruff==0.3.5
pre-commit==3.7.0