Spaces:
Runtime error
Runtime error
Miquel Farre
commited on
Commit
·
8245163
1
Parent(s):
5890f43
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')
|