liuyizhang commited on
Commit
6417b02
1 Parent(s): 6d54f1e

update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +6 -3
requirements.txt CHANGED
@@ -15,8 +15,11 @@ setuptools
15
  supervision
16
  termcolor
17
  timm
18
- torch
19
- torchvision
 
 
 
20
  yapf
21
  numba
22
  scipy
@@ -27,7 +30,7 @@ loguru
27
 
28
  lama-cleaner==1.2.3
29
 
30
- # transformers==4.32.0
31
  # transformers@git+https://github.com/huggingface/transformers.git@main
32
 
33
  mmcv==2.0.0
 
15
  supervision
16
  termcolor
17
  timm
18
+ # torch
19
+ # torchvision
20
+ torch==2.0.0
21
+ torchvision==0.15.1
22
+
23
  yapf
24
  numba
25
  scipy
 
30
 
31
  lama-cleaner==1.2.3
32
 
33
+ transformers==4.33.2
34
  # transformers@git+https://github.com/huggingface/transformers.git@main
35
 
36
  mmcv==2.0.0