From 3558b9cdcc5ef8fa88e5bf34abaf64e929fa5125 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 3 Nov 2024 17:56:03 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 1 + 1 file changed, 1 insertion(+) 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