Muhammad Taqi Raza
commited on
Commit
Β·
def28a9
1
Parent(s):
9a309cf
gradio
Browse files- 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)
|