Spaces:
Runtime error
Runtime error
Commit
·
19d2317
1
Parent(s):
b0bb6ea
Remove environment variable from flash-attn installation command for CUDA build
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ import spaces
|
|
| 10 |
import torch
|
| 11 |
import yaml
|
| 12 |
|
| 13 |
-
subprocess.run('pip install flash-attn --no-build-isolation',
|
| 14 |
|
| 15 |
# Get the PyTorch and CUDA versions
|
| 16 |
torch_version = torch.__version__.split("+")[0] # Strips any "+cuXXX" suffix
|
|
|
|
| 10 |
import torch
|
| 11 |
import yaml
|
| 12 |
|
| 13 |
+
subprocess.run('pip install flash-attn --no-build-isolation', shell=True)
|
| 14 |
|
| 15 |
# Get the PyTorch and CUDA versions
|
| 16 |
torch_version = torch.__version__.split("+")[0] # Strips any "+cuXXX" suffix
|