cleanup
Browse files
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')
|