YuE-music-generator-demo

Runtime error

KingNish commited on Aug 19

Commit

042463d

verified ·

1 Parent(s): f2f690b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import re
 print("Installing flash-attn...")
 # Install flash attention
 subprocess.run(
-    "pip install flash-attn --no-build-isolation",
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
     shell=True
 )

 print("Installing flash-attn...")
 # Install flash attention
 subprocess.run(
+    "pip install flash-attn==2.7.4.post1 --no-build-isolation",
     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
     shell=True
 )