test

requirements.txt

@@ -1,18 +1,27 @@
-torchvision
+# requirements.txt - Optimized for HF CPU Space
+
+# CPU-only PyTorch (much smaller and works on CPU Spaces)
+torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
+# Core libraries
+diffusers @ git+https://github.com/zhuang2002/diffusers_colorflow.git
 accelerate==1.0.1
-datasets==3.1.0
-git+https://github.com/zhuang2002/diffusers_colorflow.git
-einops==0.3.0
-gradio_client==1.3.0
-huggingface-hub==0.25.0
-matplotlib==3.7.5
-numpy==1.24.4
-opencv-python==4.9.0.80
+transformers==4.46.3
 peft==0.13.2
-Pillow==9.5.0
-safetensors==0.4.3
-sentencepiece==0.2.0
-torchaudio
+einops==0.8.0
+safetensors==0.4.5
+
+# UI & Utils
+gradio>=5.8.0
+spaces
+huggingface-hub==0.26.0
+Pillow==10.4.0
+opencv-python-headless==4.10.0.84
+numpy==1.26.4
 tqdm==4.67.1
-transformers==4.46.3
-webdataset==0.2.86
+
+# Others
+datasets==3.1.0
+matplotlib==3.9.2
+webdataset==0.2.86
+sentencepiece==0.2.0