multimodalart HF staff commited on
Commit
818569d
·
verified ·
1 Parent(s): 7b69ae8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,8 +4,9 @@ from typing import Union
4
  is_spaces = True if os.environ.get("SPACE_ID") else False
5
 
6
  if is_spaces:
7
- import spaces
8
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 
9
 
10
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
11
  import sys
 
4
  is_spaces = True if os.environ.get("SPACE_ID") else False
5
 
6
  if is_spaces:
 
7
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
8
+ import spaces
9
+
10
 
11
  os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
12
  import sys