From 7a581bdbe2f8113c11ac6903c41771cf327cf0c5 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 26 Mar 2025 04:13:43 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 4e9e5c5a4e..f05cc95d60 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -306,7 +306,7 @@ class FastBaseModel: # Stop SDPA for some archs like Pixtral / Mistral3 kwargs["attn_implementation"] = "sdpa" if not supports_sdpa: - print(f"Unsloth: {model_type_arch} does not support SDPA - switching to eager!") + print(f"Unsloth: {model_type_arch.title()} does not support SDPA - switching to eager!") del kwargs["attn_implementation"] pass