stojnvla commited on
Commit
a189747
·
verified ·
1 Parent(s): 264b47b

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -5
requirements.txt CHANGED
@@ -1,5 +1,5 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu117
2
- cupy-cuda11x
3
  huggingface-hub
4
  kornia
5
  kornia_rs
@@ -8,8 +8,8 @@ open_clip_torch
8
  pillow
9
  scipy
10
  timm
11
- torch==2.0.1
12
- torchvision==0.15.2
13
- torchaudio==2.0.2
14
  tqdm
15
  triton
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu121
2
+ cupy-cuda12x
3
  huggingface-hub
4
  kornia
5
  kornia_rs
 
8
  pillow
9
  scipy
10
  timm
11
+ torch
12
+ torchvision
13
+ torchaudio
14
  tqdm
15
  triton