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