Create requirements.txt

requirements.txt

@@ -0,0 +1,36 @@
+# ===== Core ML Stack =====
+# 使用PyTorch 2.x以获得更好的性能
+torch==2.5.1
+torchvision==0.20.1
+
+# Stable Diffusion + LoRA
+diffusers==0.31.0
+transformers==4.46.3
+accelerate==1.1.1
+safetensors==0.4.5
+peft==0.13.2
+compel==2.0.3
+
+# huggingface_hub版本必须<1.0以兼容transformers 4.46.3
+# 使用稳定版本避免版本冲突
+huggingface-hub>=0.23.2,<1.0
+
+# Gradio Interface
+gradio==5.5.0
+spaces==0.30.3
+
+# Core Utilities
+Pillow==11.0.0
+numpy==1.26.4
+python-dateutil==2.9.0.post0
+
+# Optional: xFormers for additional memory optimization
+# 如果环境支持，取消注释以下行
+# xformers==0.0.28.post2
+
+# For logging / error handling
+requests==2.32.3
+tqdm==4.66.5
+
+# Optional visualization/debugging
+matplotlib==3.9.2