Remove unused kwargs from model definitions (#13555)

2026-09-09 11:40:01 +08:00 · 2025-02-24 17:13:52 -08:00
parent f61528d46d
commit cdc1fa12eb
104 changed files with 436 additions and 1654 deletions
@@ -16,8 +16,6 @@ Further update the model as follows:
        self,
        input_ids: torch.Tensor,
        positions: torch.Tensor,
-        kv_caches: List[torch.Tensor],
-        attn_metadata: AttentionMetadata,
  +     pixel_values: torch.Tensor,
    ) -> SamplerOutput:
  ```