def filter_tensors(cls, item: tuple[str, Callable[[], Tensor]]) -> tuple[str, Callable[[], Tensor]] | None:
name, gen = item
- if not name.startswith("visual.") and not name.startswith("audio_tower."):
- return None
-
if name.startswith("thinker."):
name = name.replace("thinker.", "")
+ if not name.startswith("visual.") and not name.startswith("audio_tower."):
+ return None
+
if "audio_bos_eos_token" in name:
# this tensor is left unused in transformers code
# https://github.com/huggingface/transformers/blob/6e3063422c4b1c014aa60c32b9254fd2902f0f28/src/transformers/models/qwen2_5_omni/modular_qwen2_5_omni.py#L1809