feishen29 commited on
Commit
442c1bb
·
verified ·
1 Parent(s): c710d46

Upload requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -4
requirements.txt CHANGED
@@ -1,13 +1,14 @@
1
- torch==2.2.1
2
- torchvision==0.17.1
 
 
3
  numpy
4
  scipy
5
  scikit-image
6
  opencv-python
7
  pillow
8
  diffusers==0.25.1
9
- transformers==4.41.2 # 保留最新版本
10
- accelerate==0.30.1
11
  matplotlib
12
  tqdm
13
  config
 
1
+ transformers==4.36.2
2
+ torch==2.0.1
3
+ torchvision==0.15.2
4
+ torchaudio==2.0.2
5
  numpy
6
  scipy
7
  scikit-image
8
  opencv-python
9
  pillow
10
  diffusers==0.25.1
11
+ accelerate==0.26.1
 
12
  matplotlib
13
  tqdm
14
  config