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,