Flying-Lynx commited on
Commit
bb17d35
·
1 Parent(s): 72ab313

modify requirements

Browse files
Files changed (2) hide show
  1. pre-requirements.txt +3 -0
  2. requirements.txt +0 -3
pre-requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
2
+ torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
3
+ torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
requirements.txt CHANGED
@@ -35,9 +35,6 @@ safetensors==0.4.3
35
  scikit-learn==1.2.2
36
  scipy==1.11.1
37
  sentencepiece==0.1.99
38
- torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
39
- torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
40
- torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
41
  flash-attn==2.4.2
42
  timm==0.9.11
43
  toml==0.10.2
 
35
  scikit-learn==1.2.2
36
  scipy==1.11.1
37
  sentencepiece==0.1.99
 
 
 
38
  flash-attn==2.4.2
39
  timm==0.9.11
40
  toml==0.10.2