remove backend='inductor' in local patch

2025-04-07 17:05:08 -04:00
parent 04624c5a8d
commit bdaaba2784
1 changed files with 0 additions and 1 deletions
--- a/src/axolotl/monkeypatch/attention/flex_attn.py
+++ b/src/axolotl/monkeypatch/attention/flex_attn.py
@@ -40,7 +40,6 @@ def patch_flex_wrapper():
            if not self._is_flex_compiled:
                self._compiled_flex_attention = torch.compile(
                    flex_attention,
-                    backend="inductor",
                    mode="max-autotune-no-cudagraphs",
                    fullgraph=True,
                )