From fb25e956885e464cb309b53e5e92d36a59542607 Mon Sep 17 00:00:00 2001 From: Lucia Fang <116399278+luccafong@users.noreply.github.com> Date: Sat, 12 Jul 2025 14:05:32 +0800 Subject: [PATCH] [Docs] Update basic.md (#20846) --- docs/contributing/model/basic.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/docs/contributing/model/basic.md b/docs/contributing/model/basic.md index 542351fd66bb0..edd9a47e132fd 100644 --- a/docs/contributing/model/basic.md +++ b/docs/contributing/model/basic.md @@ -73,6 +73,8 @@ def forward( self, input_ids: torch.Tensor, positions: torch.Tensor, + intermediate_tensors: Optional[IntermediateTensors] = None, + inputs_embeds: Optional[torch.Tensor] = None, ) -> torch.Tensor: ... ```