[Docs] Enable fail_on_warning for the docs build in CI (#25580)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
Harry Mellor
2025-09-24 20:30:33 +01:00
committed by GitHub
parent f84a472a03
commit 8c853050e7
20 changed files with 81 additions and 87 deletions

View File

@@ -944,11 +944,10 @@ class Zamba2ForCausalLM(nn.Module, HasInnerState, IsHybrid):
hidden_states: torch.Tensor,
) -> Optional[torch.Tensor]:
"""Compute logits for next token prediction.
Args:
hidden_states: Hidden states from model forward pass
sampling_metadata: Metadata for sampling process
Returns:
Logits for next token prediction
"""