Fixed Sage detection support

2025-03-25 08:13:00 +01:00
parent 1949b61a20
commit d7fcce24c3
2 changed files with 5 additions and 3 deletions
--- a/wan/modules/attention.py
+++ b/wan/modules/attention.py
@@ -135,9 +135,9 @@ def get_supported_attention_modes():
    ret = get_attention_modes()
    if not sage_supported:
        if "sage" in ret:
-            del ret["sage"]
+            ret.remove("sage")
        if "sage2" in ret:
-            del ret["sage2"]
+            ret.remove("sage2")
    return ret

 __all__ = [