From f01757601f8f24a7be3799853d96111a6dd9f449 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 16 Mar 2025 22:10:36 -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 24015f82fe..5a990566a7 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -289,6 +289,7 @@ class FastBaseModel: # Cannot be None, since HF now checks for the config if load_in_4bit: kwargs["quantization_config"] = bnb_config + print(model_name) model = auto_model.from_pretrained( model_name, device_map = device_map,