From f92544d75a24b28aa7086d865a81ca102e66e683 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 15 Mar 2025 17:58:14 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 69cc1e6884..bcd23ee490 100644 --- 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