SUPERSHANKY commited on
Commit
4ccfec7
1 Parent(s): 3fcf59e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu118
2
  addict==2.4.0
3
  albumentations==1.3.0
4
  einops==0.6.0
@@ -15,8 +15,8 @@ prettytable==3.6.0
15
  pytorch-lightning
16
  safetensors==0.3.0
17
  timm
18
- https://download.pytorch.org/whl/cu118/torch-1.13.1%2Bcu118-cp39-cp39-linux_x86_64.whl
19
- torchvision==0.14.1+cu118
20
  fastai
21
  transformers==4.27.1
22
  datasets==2.9.0
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu117
2
  addict==2.4.0
3
  albumentations==1.3.0
4
  einops==0.6.0
 
15
  pytorch-lightning
16
  safetensors==0.3.0
17
  timm
18
+ https://download.pytorch.org/whl/cu118/torch-1.13.1%2Bcu117-cp39-cp39-linux_x86_64.whl
19
+ torchvision==0.14.1+cu117
20
  fastai
21
  transformers==4.27.1
22
  datasets==2.9.0