[Chore] Update type annotation of input_ids in model forward (#33063)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2026-01-26 22:02:10 +08:00
committed by GitHub
parent f4a0921c9c
commit dcd80206b7
164 changed files with 243 additions and 241 deletions

View File

@@ -998,7 +998,7 @@ class Qwen3MoeLLMModel(Qwen3MoeModel):
def forward(
self,
input_ids: torch.Tensor,
input_ids: torch.Tensor | None,
positions: torch.Tensor,
intermediate_tensors: IntermediateTensors | None = None,
inputs_embeds: torch.Tensor | None = None,
@@ -1819,7 +1819,7 @@ class Qwen3OmniMoeThinkerForConditionalGeneration(
def forward(
self,
input_ids: torch.Tensor,
input_ids: torch.Tensor | None,
positions: torch.Tensor,
intermediate_tensors: IntermediateTensors | None = None,
inputs_embeds: torch.Tensor | None = None,