Update _utils.py

2025-03-15 17:58:14 -07:00 · 2025-03-15 17:58:14 -07:00 · f92544d75a
commit f92544d75a
parent 7f2400da25
1 changed files with 3 additions and 0 deletions
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@ -1204,6 +1204,9 @@ def unsloth_compile_transformers(
            return_logits          = return_logits,
        )
    pass
+    # Redo patches which override compiler
+    for temporary_patch in TEMPORARY_PATCHES:
+        temporary_patch()
    return model_types
 pass