yhzx233 commited on
Commit
aede480
·
1 Parent(s): 0e5ec8c

fix: install flash attn

Browse files
Files changed (2) hide show
  1. app.py +4 -0
  2. requirements.txt +1 -2
app.py CHANGED
@@ -1,3 +1,7 @@
 
 
 
 
1
  import gradio as gr
2
  import torch
3
  import torchaudio
 
1
+ import sys
2
+ import subprocess
3
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "flash-attn"])
4
+
5
  import gradio as gr
6
  import torch
7
  import torchaudio
requirements.txt CHANGED
@@ -13,5 +13,4 @@ requests
13
  openai
14
  PyYAML
15
  einops
16
- huggingface_hub
17
- flash-attn
 
13
  openai
14
  PyYAML
15
  einops
16
+ huggingface_hub