Guanzheng commited on
Commit
6912511
·
1 Parent(s): 94416a7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -2,5 +2,6 @@ torch==2.0.1
2
  tqdm==4.64.1
3
  transformers==4.29.1
4
  torchdiffeq==0.2.3
 
5
  PyPDF2
6
  einops
 
2
  tqdm==4.64.1
3
  transformers==4.29.1
4
  torchdiffeq==0.2.3
5
+ flash-attn==2.3.2 --no-build-isolation
6
  PyPDF2
7
  einops