byte-vortex's picture
Create requirements.txt
5d7b9b7 verified
# === Core Frameworks & MCP ===
gradio[mcp]>=4.0.0 # Required for the UI and enabling the MCP Server endpoint
torch>=2.0.0 # PyTorch, the underlying ML framework
diffusers>=0.20.0 # Required for the StableDiffusionImg2ImgPipeline
transformers>=4.30.0 # Required for the text encoder (CLIP)
# === Image Processing & Utilities ===
Pillow>=9.0.0 # Required for PIL.Image operations
numpy>=1.20.0 # General numerical library
# === Optional (But Recommended for Performance/Compliance) ===
# NOTE: These are crucial for fast performance on GPU hardware.
accelerate # For easy float16/bfloat16 and multi-GPU usage
xformers # For memory-efficient attention (faster generation)
invisible-watermark # For compliance/tagging of generated images