riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update attention_visualizer.py (#37860)

Browse Source
This commit is contained in:
Tanuj Rai 2025-06-24 19:51:36 +05:30 committed by GitHub
parent 9f42c1f192
commit 719058c625
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/utils/attention_visualizer.py
View File

@ -151,7 +151,7 @@ class AttentionMaskVisualizer:
config = AutoConfig.from_pretrained(model_name)
self.image_token = "<img>"
if hasattr(config.get_text_config(), "sliding_window"):
config.sliding_window = 5
self.sliding_window = getattr(config.get_text_config(), "sliding_window", None)
try:
mapped_cls = _get_model_class(config, MODEL_MAPPING)
except Exception: