[5/N] Initialize MM components in context managers (Q-Z) (#32695)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2026-01-21 03:10:23 +08:00
committed by GitHub
parent f0feb1cf81
commit 193069d129
9 changed files with 178 additions and 168 deletions

View File

@@ -620,7 +620,6 @@ class RadioInternVisionModel(nn.Module):
x: torch.Tensor,
imgs_sizes: torch.Tensor | None = None,
) -> torch.FloatTensor:
assert self.patch_generator is not None
hidden_states = self.patch_generator(x, imgs_sizes=imgs_sizes)
attn_mask = None
if imgs_sizes is not None and len(imgs_sizes) > 1: