chaojiemao commited on
Commit
227e14f
·
verified ·
1 Parent(s): 4ba6875

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ import threading
14
  import spaces
15
 
16
  subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
17
- subprocess.run(shlex.split('pip install flash-attn==2.5.8 --no-build-isolation'),
18
  env=os.environ | {'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"})
19
  subprocess.run(shlex.split('pip install scepter'))
20
 
 
14
  import spaces
15
 
16
  subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
17
+ subprocess.run(shlex.split('pip install flash-attn --no-build-isolation'),
18
  env=os.environ | {'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"})
19
  subprocess.run(shlex.split('pip install scepter'))
20