hpoghos commited on
Commit
bc9ad47
·
verified ·
1 Parent(s): 25c1883

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -3
requirements.txt CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  accelerate==0.28.0
2
  bitsandbytes==0.43.0
3
  transformers==4.39.3
@@ -22,9 +26,6 @@ scikit-image==0.20.0
22
  scikit-learn==1.2.2
23
  scipy==1.9.1
24
  seaborn==0.12.2
25
- torch
26
- torchdata
27
- torchvision
28
  modelscope==1.13.3
29
  tqdm==4.65.0
30
  xformers==0.0.19
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
+ torch
3
+ torchdata
4
+ torchvision
5
  accelerate==0.28.0
6
  bitsandbytes==0.43.0
7
  transformers==4.39.3
 
26
  scikit-learn==1.2.2
27
  scipy==1.9.1
28
  seaborn==0.12.2
 
 
 
29
  modelscope==1.13.3
30
  tqdm==4.65.0
31
  xformers==0.0.19