Muhammad Taqi Raza commited on
Commit
def28a9
Β·
1 Parent(s): 9a309cf
Files changed (1) hide show
  1. requirements.txt +10 -8
requirements.txt CHANGED
@@ -1,8 +1,10 @@
1
  # ─── Core DL ─────────────────────────────────────────────────────────
2
- torch==2.3.1
3
- torchvision==0.18.1
 
 
4
  diffusers==0.32.2
5
- accelerate>=1.1.1
6
  transformers>=4.46.2
7
  numpy==1.26.0
8
  einops
@@ -14,18 +16,18 @@ sentencepiece>=0.2.0
14
  opencv-python-headless>=4.10.0.84
15
  imageio>=2.35.1
16
  imageio-ffmpeg>=0.5.1
17
- decord
18
  scipy
19
  scikit-image
20
  scikit-video>=1.1.11
21
  moviepy==2.2.1
22
 
23
  # ─── Specialized libs ────────────────────────────────────────────────
24
- SwissArmyTransformer>=0.4.12
25
- spandrel>=0.4.0
26
  gdown
27
- qwen_vl_utils
28
  openai>=1.54.0
29
 
30
  # ─── UI ──────────────────────────────────────────────────────────────
31
- gradio
 
1
  # ─── Core DL ─────────────────────────────────────────────────────────
2
+ torch==2.3.1+cpu # Force CPU version
3
+ torchvision==0.18.1+cpu # Same here
4
+ --extra-index-url https://download.pytorch.org/whl/cpu
5
+
6
  diffusers==0.32.2
7
+ accelerate>=1.1.1 # Required for diffusers; ZeroGPU-friendly when `device_map="cpu"`
8
  transformers>=4.46.2
9
  numpy==1.26.0
10
  einops
 
16
  opencv-python-headless>=4.10.0.84
17
  imageio>=2.35.1
18
  imageio-ffmpeg>=0.5.1
19
+ decord # ⚠ Optional, may fail on some CPUs β€” fallback to OpenCV if needed
20
  scipy
21
  scikit-image
22
  scikit-video>=1.1.11
23
  moviepy==2.2.1
24
 
25
  # ─── Specialized libs ────────────────────────────────────────────────
26
+ SwissArmyTransformer>=0.4.12 # ⚠ Heavy β€” make sure it doesn't try to use CUDA internally
27
+ spandrel>=0.4.0 # If fails, consider making it optional
28
  gdown
29
+ qwen_vl_utils # ⚠ If Qwen2.5-VL model used, not ZeroGPU-compatible β€” disable
30
  openai>=1.54.0
31
 
32
  # ─── UI ──────────────────────────────────────────────────────────────
33
+ gradio==4.29.0 # βœ… Stable, ZeroGPU-ready (as long as backend models are CPU-safe)