jallenjia commited on
Commit
841edd1
·
1 Parent(s): 1de9b2b

adapt h200 gpu

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -4
requirements.txt CHANGED
@@ -1,13 +1,11 @@
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
@@ -20,4 +18,5 @@ av
20
  fvcore
21
  cloudpickle
22
  omegaconf
23
- pycocotools
 
 
1
  transformers==4.36.2
 
 
 
2
  numpy==1.24.4
3
  scipy==1.10.1
4
  scikit-image==0.21.0
5
  opencv-python==4.7.0.72
6
  pillow==9.4.0
7
  diffusers==0.25.0
8
+ huggingface_hub==0.25.0
9
  transformers==4.36.2
10
  accelerate==0.26.1
11
  matplotlib==3.7.4
 
18
  fvcore
19
  cloudpickle
20
  omegaconf
21
+ pycocotools
22
+ pydantic==2.10.6