Mountchicken commited on
Commit
faab470
·
verified ·
1 Parent(s): b86cd47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -2,7 +2,8 @@ import argparse
2
  import json
3
  import os
4
  os.system("pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124")
5
- os.system("pip install flash-attn==2.5.2")
 
6
  import sys
7
  import threading
8
 
 
2
  import json
3
  import os
4
  os.system("pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124")
5
+ import subprocess
6
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
7
  import sys
8
  import threading
9