Beepeen24 commited on
Commit
b18069f
·
verified ·
1 Parent(s): 0bcd072

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +16 -3
requirements.txt CHANGED
@@ -1,17 +1,30 @@
1
- torch==2.9.1
2
- torchvision==0.24.1
 
3
  gradio==3.44.0
4
  diffusers==0.24.0
5
  transformers==4.31.0
6
  accelerate==0.24.0
7
  safetensors==0.4.0
 
 
8
  opencv-python-headless==4.8.0.74
9
  Pillow==9.5.0
10
  numpy==1.24.3
 
 
11
  huggingface-hub==0.20.0
12
  omegaconf==2.3.0
13
  einops==0.7.0
 
 
14
  insightface==0.7.3
 
 
15
  controlnet_aux==0.0.7
 
 
16
  pydantic==2.10.6
17
- xformers==0.0.25
 
 
 
1
+ # NO PyTorch or torchvision - they're pre-installed on ZeroGPU!
2
+
3
+ # Core dependencies
4
  gradio==3.44.0
5
  diffusers==0.24.0
6
  transformers==4.31.0
7
  accelerate==0.24.0
8
  safetensors==0.4.0
9
+
10
+ # Computer vision
11
  opencv-python-headless==4.8.0.74
12
  Pillow==9.5.0
13
  numpy==1.24.3
14
+
15
+ # AI/ML utilities
16
  huggingface-hub==0.20.0
17
  omegaconf==2.3.0
18
  einops==0.7.0
19
+
20
+ # Face recognition
21
  insightface==0.7.3
22
+
23
+ # ControlNet
24
  controlnet_aux==0.0.7
25
+
26
+ # Data validation
27
  pydantic==2.10.6
28
+
29
+ # Optional optimizations (comment out if causing issues)
30
+ # xformers==0.0.25