Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -109,7 +109,7 @@ def run_gradio_ui():
|
|
109 |
|
110 |
def main():
|
111 |
import subprocess
|
112 |
-
|
113 |
if os.path.isfile(CHECKPOINT_FP8_FILE):
|
114 |
print("β
Model checkpoint already exists. Skipping download.")
|
115 |
else:
|
|
|
109 |
|
110 |
def main():
|
111 |
import subprocess
|
112 |
+
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
113 |
if os.path.isfile(CHECKPOINT_FP8_FILE):
|
114 |
print("β
Model checkpoint already exists. Skipping download.")
|
115 |
else:
|