favoredone commited on
Commit
fc86340
·
verified ·
1 Parent(s): 404886a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -12,11 +12,6 @@ import time
12
  import urllib.parse
13
 
14
  # Attempt to install flash-attn
15
- try:
16
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, check=True, shell=True)
17
- except subprocess.CalledProcessError as e:
18
- print(f"Error installing flash-attn: {e}")
19
- print("Continuing without flash-attn.")
20
 
21
  # Determine the device to use
22
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
12
  import urllib.parse
13
 
14
  # Attempt to install flash-attn
 
 
 
 
 
15
 
16
  # Determine the device to use
17
  device = "cuda" if torch.cuda.is_available() else "cpu"