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

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2026-01-26 22:02:10 +08:00
parent f4a0921c9c
commit dcd80206b7
164 changed files with 243 additions and 241 deletions
--- a/docs/contributing/model/basic.md
+++ b/docs/contributing/model/basic.md
@@ -71,7 +71,7 @@ class MyModel(nn.Module):
 ```python
 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,