diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py
index 0eade901c7..d974f76a89 100644
--- a/unsloth/models/loader.py
+++ b/unsloth/models/loader.py
@@ -625,7 +625,8 @@ class FastModel(FastBaseModel):
         is_vlm = (x.endswith("ForConditionalGeneration") for x in model_config.architectures)
         is_vlm = is_vlm or hasattr(model_config, "vision_config")
         auto_model = AutoModelForVision2Seq if is_vlm else AutoModelForCausalLM
-
+        print(auto_model)
+        print(is_vlm)
         model, tokenizer = FastBaseModel.from_pretrained(
             model_name        = model_name,
             max_seq_length    = max_seq_length,