spdraptor commited on
Commit
24cccae
·
verified ·
1 Parent(s): 27ca726

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  #Script added by SPDraptor
2
  import spaces
3
  from typing import Optional
4
- # import subprocess
5
- # subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  import torch
7
  print("cuda present = ",torch.cuda.is_available())
8
  import os
 
1
  #Script added by SPDraptor
2
  import spaces
3
  from typing import Optional
4
+ import subprocess
5
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
6
  import torch
7
  print("cuda present = ",torch.cuda.is_available())
8
  import os