SyrWin commited on
Commit
f47b963
·
1 Parent(s): 3a904b0

updated requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +0 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import torch
3
  import argparse
 
1
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
2
  import os
3
  import torch
4
  import argparse
requirements.txt CHANGED
@@ -4,7 +4,6 @@ torch
4
  pytorch_lightning
5
  torch_geometric
6
  transformers
7
- flash_attn
8
  peft
9
  nltk
10
  rouge_score
 
4
  pytorch_lightning
5
  torch_geometric
6
  transformers
 
7
  peft
8
  nltk
9
  rouge_score