Spaces:
Running
Running
Abhishek Thakur
commited on
Commit
·
4d6ffe3
1
Parent(s):
89add76
install flash attn
Browse files
app.py
CHANGED
|
@@ -4,6 +4,9 @@ is_spaces = True if os.environ.get("SPACE_ID") else False
|
|
| 4 |
|
| 5 |
if is_spaces:
|
| 6 |
import spaces
|
|
|
|
|
|
|
|
|
|
| 7 |
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
| 8 |
import sys
|
| 9 |
|
|
|
|
| 4 |
|
| 5 |
if is_spaces:
|
| 6 |
import spaces
|
| 7 |
+
|
| 8 |
+
# install flash-attention
|
| 9 |
+
subprocess.run(["pip", "install", "flash-attention", "--no-build-isolation"])
|
| 10 |
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
|
| 11 |
import sys
|
| 12 |
|