Spaces:

autotrain-projects
/

train-flux-lora-ease

Running on CPU Upgrade

abhishek HF staff commited on Aug 28

Commit

4d6ffe3

•

1 Parent(s): 89add76

install flash attn

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,9 @@ is_spaces = True if os.environ.get("SPACE_ID") else False
 if is_spaces:
     import spaces
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 import sys

 if is_spaces:
     import spaces
+    # install flash-attention
+    subprocess.run(["pip", "install", "flash-attention", "--no-build-isolation"])
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 import sys