Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from pipelines.hidream_image.pipeline_hidream_image import HiDreamImagePipeline
|
|
9 |
import subprocess
|
10 |
|
11 |
print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
|
12 |
-
subprocess.run('pip install flash-attn --no-build-isolation',
|
13 |
|
14 |
from nf4 import *
|
15 |
|
|
|
9 |
import subprocess
|
10 |
|
11 |
print(subprocess.check_output(["nvcc", "--version"]).decode("utf-8"))
|
12 |
+
subprocess.run('python -m pip install flash-attn --no-build-isolation', shell=True)
|
13 |
|
14 |
from nf4 import *
|
15 |
|