diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 5524d8f16d..5944143431 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -441,6 +441,7 @@ class FastBaseModel: if do_forced_float32: torch_dtype = torch.bfloat16 raise_handler = RaiseUninitialized() + print(model_name, kwargs) model = auto_model.from_pretrained( model_name, device_map = device_map,