Malaji71 commited on
Commit
b6b9801
verified
1 Parent(s): b661d19

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +20 -9
requirements.txt CHANGED
@@ -1,26 +1,37 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
 
 
2
 
3
- # First install PyTorch from CUDA index
4
- torch==2.1.0+cu118
5
- torchvision==0.16.0+cu118
6
-
7
- # Then install everything else
8
  gradio==5.33.2
 
 
9
  transformers==4.44.0
10
  accelerate==0.24.1
 
 
11
  pillow==10.0.1
12
  requests==2.31.0
13
  numpy==1.26.4
 
 
 
 
14
  sentence-transformers==2.2.2
 
 
15
  einops==0.8.0
16
  timm==1.0.3
17
  opencv-python==4.8.1.78
18
  opencv-contrib-python==4.8.1.78
19
  mediapipe==0.10.7
20
  retina-face==0.0.13
21
- gdown==4.7.1
22
- tqdm==4.66.1
23
  deepface==0.0.79
 
 
24
  tf-keras==2.15.0
25
  tensorflow==2.17.0
26
- flash-attn==2.7.4.post1 --no-build-isolation
 
 
1
+ # PyTorch y torchvision versi贸n CPU
2
+ torch==2.1.0
3
+ torchvision==0.16.0
4
 
5
+ # Gradio para interfaz
 
 
 
 
6
  gradio==5.33.2
7
+
8
+ # Transformers y aceleraci贸n
9
  transformers==4.44.0
10
  accelerate==0.24.1
11
+
12
+ # Utilidades comunes
13
  pillow==10.0.1
14
  requests==2.31.0
15
  numpy==1.26.4
16
+ tqdm==4.66.1
17
+ gdown==4.7.1
18
+
19
+ # Procesamiento de lenguaje
20
  sentence-transformers==2.2.2
21
+
22
+ # Modelado visual y visi贸n por computador
23
  einops==0.8.0
24
  timm==1.0.3
25
  opencv-python==4.8.1.78
26
  opencv-contrib-python==4.8.1.78
27
  mediapipe==0.10.7
28
  retina-face==0.0.13
29
+
30
+ # Deepface (requiere solo CPU si est谩 bien configurado)
31
  deepface==0.0.79
32
+
33
+ # Tensorflow para modelos compatibles con CPU
34
  tf-keras==2.15.0
35
  tensorflow==2.17.0
36
+
37
+ # 鈿狅笍 IMPORTANTE: NO INCLUIMOS flash-attn, requiere GPU