frogleo commited on
Commit
e7fd8db
·
verified ·
1 Parent(s): 0022dfc

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -5,9 +5,9 @@ diffusers
5
  einops
6
  opencv-python
7
  numpy
8
- torch
9
- transformers
10
- torchvision
11
  #taming-transformers-rom1504
12
  #ConfigArgParse
13
  #ipdb
 
5
  einops
6
  opencv-python
7
  numpy
8
+ torch==2.4.0
9
+ transformers==4.37.0
10
+ torchvision==0.19.0
11
  #taming-transformers-rom1504
12
  #ConfigArgParse
13
  #ipdb