From 6218eae6add593a3f31bd1b5e7e788692f83058b Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 12 Mar 2025 21:38:51 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 8bfb9a119a..f4be016770 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -39,6 +39,7 @@ from unsloth_zoo.peft_utils import ( SKIP_QUANTIZATION_MODULES, requires_grad_for_gradient_checkpointing, ) +from transformers.models.llama.modeling_llama import logger from transformers import __version__ as transformers_version from triton import __version__ as triton_version from unsloth_zoo.utils import _get_dtype