From 9bc273b9500666bb7367e7bfc167a5a5c06e1cb2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Fri, 14 Mar 2025 04:59:29 -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 dbeb487e0d..05b7489eae 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -205,7 +205,7 @@ class FastBaseModel: global FORCE_EAGER_ATTENTION attn_implementation = "sdpa" - for disable_sdpa_name in FORCE_EAGER_ATTENTION: + for disable_name in FORCE_EAGER_ATTENTION: if (disable_name.lower() == model_type_arch.lower() or \ disable_name.lower() in model_name.lower()):