[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:
@@ -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
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user