diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 8dca483783..38c3c1b159 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -408,6 +408,7 @@ class FastBaseModel: torch_dtype = dtype if do_forced_float32: torch_dtype = torch.bfloat16 + print(kwargs) model = auto_model.from_pretrained( model_name, device_map = device_map,