From 702f85bd5411f5d36b53b91563336368b6931bee Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 12 Mar 2025 21:34:34 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index f564c30a9f..490938ae56 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -30,7 +30,9 @@ from ..kernels import ( ) from ._utils import __version__ from ._utils import * +from ..save import patch_saving_functions from peft import LoraConfig, TaskType, get_peft_model as _get_peft_model +from peft import PeftModelForCausalLM from transformers import set_seed as transformers_set_seed from unsloth_zoo.peft_utils import ( get_peft_regex, @@ -48,7 +50,7 @@ import gc import math import functools from typing import Optional, Tuple, List, Union -import re, os, inspect, math, sys +import re, inspect, sys import types try: from huggingface_hub.utils import get_token