From 82fa68ca14c1c07b9d8848c2ec70b00a557d0f60 Mon Sep 17 00:00:00 2001 From: 5pipeline <37110366+IceGiraffe@users.noreply.github.com> Date: Wed, 4 Jun 2025 20:20:31 +0800 Subject: [PATCH] fix(attention_visualizer): add default value for image_seq_length (#38577) --- src/transformers/utils/attention_visualizer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/transformers/utils/attention_visualizer.py b/src/transformers/utils/attention_visualizer.py index ac52dd46ce4..2be4a72e7e2 100644 --- a/src/transformers/utils/attention_visualizer.py +++ b/src/transformers/utils/attention_visualizer.py @@ -178,6 +178,7 @@ class AttentionMaskVisualizer: def visualize_attention_mask(self, input_sentence: str, suffix=""): model = self.model kwargs = {} + image_seq_length = None if self.config.model_type in PROCESSOR_MAPPING_NAMES: img = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/bee.jpg?download=true" img = Image.open(requests.get(img, stream=True).raw)