diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index ba6392eab8..091dbaee29 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -48,6 +48,7 @@ __all__ = [ "patch_regional_compilation", "patch_layernorm", "patch_torch_compile", + "patch_model_and_tokenizer", ] import torch