Yaron Koresh commited on
Commit
84f9d3c
·
verified ·
1 Parent(s): b525741

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -6,8 +6,8 @@ transformers
6
  pathos
7
  dill
8
  --extra-index-url https://download.pytorch.org/whl/cu121
9
- torch==2.1.0
10
- torchaudio>=2.1.0
11
- torchvision>=0.16.0
12
  triton
13
  xformers
 
6
  pathos
7
  dill
8
  --extra-index-url https://download.pytorch.org/whl/cu121
9
+ torch
10
+ torchaudio
11
+ torchvision
12
  triton
13
  xformers