Nymbo commited on
Commit
795d073
ยท
verified ยท
1 Parent(s): ba6ca56

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +26 -15
requirements.txt CHANGED
@@ -1,23 +1,34 @@
1
- transformers==4.36.2
2
  torch==2.0.1
3
  torchvision==0.15.2
4
  torchaudio==2.0.2
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.25.0
11
  transformers==4.36.2
12
  accelerate==0.26.1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
13
  matplotlib==3.7.4
14
  tqdm==4.64.1
15
- config==0.5.1
16
  einops==0.7.0
17
- onnxruntime==1.16.2
18
- basicsr
19
- av
20
- fvcore
21
- cloudpickle
22
- omegaconf
23
- pycocotools
 
 
1
+ # Core ML Libraries
2
  torch==2.0.1
3
  torchvision==0.15.2
4
  torchaudio==2.0.2
 
 
 
 
 
 
5
  transformers==4.36.2
6
  accelerate==0.26.1
7
+ diffusers==0.25.0
8
+ huggingface_hub<0.15.0 # Pin huggingface_hub below v0.15 for diffusers 0.25.0 compatibility
9
+
10
+ # Image/Video/Array Processing
11
+ numpy==1.24.4
12
+ Pillow==9.4.0 # PIL is an alias for Pillow installed this way
13
+ opencv-python==4.7.0.72
14
+ scikit-image==0.21.0
15
+ scipy==1.10.1
16
+ av # For video processing if needed by dependencies
17
+
18
+ # ONNX
19
+ onnxruntime==1.16.2
20
+ onnx
21
+
22
+ # Utilities & Frameworks
23
+ gradio # Added: required by app.py
24
  matplotlib==3.7.4
25
  tqdm==4.64.1
 
26
  einops==0.7.0
27
+ fvcore # Required by detectron2 (vendored)
28
+ pycocotools # Required by detectron2/COCO eval (vendored)
29
+ omegaconf # Required by detectron2 (vendored)
30
+ cloudpickle # Required by detectron2 (vendored)
31
+
32
+ # Misc - check if these are truly needed or are dependencies of others
33
+ config==0.5.1 # Be careful with this generic name, ensure it's the correct package
34
+ basicsr