DHEIVER commited on
Commit
c49e746
1 Parent(s): 400aec7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +38 -24
requirements.txt CHANGED
@@ -1,32 +1,46 @@
1
  # Core dependencies
2
- gradio>=4.19.2
3
- torch>=2.2.0
4
- transformers>=4.37.2
5
- diffusers>=0.25.0
6
- accelerate>=0.27.2
 
 
7
 
8
- # Image processing
9
- Pillow>=10.2.0
10
- imageio>=2.33.1
11
- opencv-python>=4.9.0.80
 
 
 
 
12
 
13
- # NLP and translation
14
- sacremoses>=0.1.1
15
- sentencepiece>=0.2.0
16
- protobuf>=4.25.2
17
- tokenizers>=0.15.2
18
 
19
  # Deep learning utilities
20
- numpy>=1.26.4
21
- scipy>=1.12.0
22
- tqdm>=4.66.2
 
 
23
 
24
- # ModelScope dependencies
25
- addict>=2.4.0
26
- modelscope>=1.12.0
27
- moviepy>=1.0.3
28
 
29
  # Additional utilities
30
- requests>=2.31.0
31
- packaging>=23.2
32
- huggingface-hub>=0.20.3
 
 
 
 
 
 
 
 
 
 
1
  # Core dependencies
2
+ gradio==4.19.2
3
+ torch==2.2.0
4
+ torchvision==0.17.0
5
+ torchaudio==2.2.0
6
+ transformers==4.37.2
7
+ diffusers==0.25.0
8
+ accelerate==0.27.2
9
 
10
+ # NLP tools
11
+ spacy==3.7.2
12
+ sacremoses==0.0.53
13
+ sentencepiece==0.1.99
14
+ protobuf==4.25.2
15
+ tokenizers==0.15.2
16
+ ftfy==6.1.3
17
+ regex==2023.12.25
18
 
19
+ # Image processing
20
+ Pillow==10.2.0
21
+ opencv-python==4.9.0.80
 
 
22
 
23
  # Deep learning utilities
24
+ numpy==1.26.4
25
+ scipy==1.12.0
26
+ tqdm==4.66.2
27
+ sympy==1.12
28
+ networkx==3.2.1
29
 
30
+ # HuggingFace specific
31
+ huggingface-hub==0.20.3
32
+ safetensors==0.4.2
 
33
 
34
  # Additional utilities
35
+ requests==2.31.0
36
+ packaging==23.2
37
+ filelock==3.13.1
38
+ psutil==5.9.8
39
+ typing_extensions==4.9.0
40
+ pyyaml==6.0.1
41
+
42
+ # Optional - for better performance with NVIDIA GPUs
43
+ --extra-index-url https://download.pytorch.org/whl/cu121
44
+ torch==2.2.0+cu121
45
+ torchvision==0.17.0+cu121
46
+ torchaudio==2.2.0+cu121