Create requirements.txt
Browse files- requirements.txt +15 -0
requirements.txt
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# === Core Frameworks & MCP ===
|
| 2 |
+
gradio[mcp]>=4.0.0 # Required for the UI and enabling the MCP Server endpoint
|
| 3 |
+
torch>=2.0.0 # PyTorch, the underlying ML framework
|
| 4 |
+
diffusers>=0.20.0 # Required for the StableDiffusionImg2ImgPipeline
|
| 5 |
+
transformers>=4.30.0 # Required for the text encoder (CLIP)
|
| 6 |
+
|
| 7 |
+
# === Image Processing & Utilities ===
|
| 8 |
+
Pillow>=9.0.0 # Required for PIL.Image operations
|
| 9 |
+
numpy>=1.20.0 # General numerical library
|
| 10 |
+
|
| 11 |
+
# === Optional (But Recommended for Performance/Compliance) ===
|
| 12 |
+
# NOTE: These are crucial for fast performance on GPU hardware.
|
| 13 |
+
accelerate # For easy float16/bfloat16 and multi-GPU usage
|
| 14 |
+
xformers # For memory-efficient attention (faster generation)
|
| 15 |
+
invisible-watermark # For compliance/tagging of generated images
|