Basic sageattn3 support

Probably going to need further timestep scheduling as the quality of sage3 on it's own is pretty bad.
2025-07-25 20:48:34 +03:00
parent 76750eec05
commit a35eb7dfa4
4 changed files with 60 additions and 38 deletions
@@ -717,6 +717,7 @@ class WanVideoModelLoader:
                    "flash_attn_2",
                    "flash_attn_3",
                    "sageattn",
+                    "sageattn_3",
                    "flex_attention",
                    "radial_sage_attention",
                    ], {"default": "sdpa"}),