Flying-Lynx commited on
Commit
90fe35e
·
1 Parent(s): 02da275

update requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -38,6 +38,7 @@ 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
  timm==0.9.11
42
  toml==0.10.2
43
  tqdm==4.66.1
 
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 --no-build-isolation
42
  timm==0.9.11
43
  toml==0.10.2
44
  tqdm==4.66.1