From 8a109a6c9f36de88d4877bb6c78d98808d297488 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 7 Aug 2025 16:21:25 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 0bae1ee10e..a9e3573bef 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -435,7 +435,7 @@ class FastBaseModel: if do_forced_float32: torch_dtype = torch.bfloat16 raise_handler = RaiseUninitialized() - print(kwargs) + print(kwargs, model_name) model = auto_model.from_pretrained( model_name, device_map = device_map, @@ -446,6 +446,7 @@ class FastBaseModel: # attn_implementation = attn_implementation, **kwargs, ) + print(model) raise_handler.remove() # Return old flag os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = old_hf_transfer