Spaces:

Dovakiins
/

qwerrwe

Build error

casperhansen commited on Dec 7, 2023

Commit

06ae392

unverified ·

1 Parent(s): a581e9f

Pin flash-attn to 2.3.3 (#919)

Files changed (1) hide show

setup.py CHANGED Viewed

@@ -46,7 +46,7 @@ setup(
     dependency_links=dependency_links,
     extras_require={
         "flash-attn": [
-            "flash-attn>=2.3.0",
         ],
         "deepspeed": [
             "deepspeed",

     dependency_links=dependency_links,
     extras_require={
         "flash-attn": [
+            "flash-attn==2.3.3",
         ],
         "deepspeed": [
             "deepspeed",