Spaces:
BAAI
/
Running on L40S

ryanzhangfan commited on
Commit
f4816fe
1 Parent(s): 023c8b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import subprocess
2
- subprocess.run("pip install gradio==4.44.0", shell=True)
3
- subprocess.run(
4
- "pip install flash-attn --no-build-isolation",
5
- env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
- shell=True,
7
- )
8
 
9
 
10
  from PIL import Image
 
1
  import subprocess
2
+ subprocess.run("FLASH_ATTNTION_SKIP_CUDA_BUILD=TRUE pip install gradio==4.44.0 flash-attn --no-build-isolation", shell=True)
3
+ # subprocess.run(
4
+ # "pip install flash-attn --no-build-isolation",
5
+ # env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
6
+ # shell=True,
7
+ # )
8
 
9
 
10
  from PIL import Image