rahul7star commited on
Commit
fb583c4
Β·
verified Β·
1 Parent(s): a097fff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ def run_gradio_ui():
109
 
110
  def main():
111
  import subprocess
112
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
113
  if os.path.isfile(CHECKPOINT_FP8_FILE):
114
  print("βœ… Model checkpoint already exists. Skipping download.")
115
  else:
 
109
 
110
  def main():
111
  import subprocess
112
+ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
113
  if os.path.isfile(CHECKPOINT_FP8_FILE):
114
  print("βœ… Model checkpoint already exists. Skipping download.")
115
  else: