mfarre HF staff commited on
Commit
8245163
·
1 Parent(s): 5890f43
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -402,7 +402,7 @@ def create_ui(examples_path: str, model_path: str):
402
  return app
403
 
404
  if __name__ == "__main__":
405
- subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
406
 
407
  # Initialize CUDA
408
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
402
  return app
403
 
404
  if __name__ == "__main__":
405
+ # subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
406
 
407
  # Initialize CUDA
408
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')