Nymbo commited on
Commit
6c9b75d
ยท
verified ยท
1 Parent(s): 895d3f6

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +23 -15
requirements.txt CHANGED
@@ -1,23 +1,31 @@
1
- transformers==4.36.2
 
 
2
  torch
3
  torchvision
4
  torchaudio
5
- numpy==1.24.4
6
- scipy==1.10.1
7
- scikit-image==0.21.0
8
- opencv-python==4.7.0.72
9
- pillow==9.4.0
10
- diffusers==0.27.2
11
- accelerate==0.26.1
12
- matplotlib==3.7.4
13
- tqdm==4.64.1
14
- config==0.5.1
 
 
 
 
 
15
  einops==0.7.0
16
- onnxruntime==1.16.2
 
 
 
17
  basicsr
18
  av
19
  fvcore
20
  cloudpickle
21
- omegaconf
22
- pycocotools
23
- huggingface_hub
 
1
+ # main/requirements.txt
2
+
3
+ # --- Core ML/DL Libraries ---
4
  torch
5
  torchvision
6
  torchaudio
7
+ transformers==4.38.2
8
+ diffusers==0.27.2 # Upgraded from 0.25.0
9
+ accelerate==0.28.0 # Upgraded from 0.26.1
10
+ huggingface_hub # Un-pinning the version to let diffusers choose
11
+
12
+ # --- Data & Image Processing ---
13
+ numpy==1.26.4
14
+ scipy==1.11.4
15
+ scikit-image==0.22.0
16
+ opencv-python==4.8.0.76
17
+ Pillow==10.1.0
18
+
19
+ # --- Utilities & Others ---
20
+ matplotlib==3.8.2
21
+ tqdm==4.66.1
22
  einops==0.7.0
23
+ omegaconf
24
+ # The following probably don't need version pins
25
+ config
26
+ onnxruntime
27
  basicsr
28
  av
29
  fvcore
30
  cloudpickle
31
+ pycocotools