[Docs] Fix confusing docstring indentation in nemotron_h.py (#49781)
Signed-off-by: Johnny-Liou <a897111@gmail.com>
This commit is contained in:
@@ -771,8 +771,7 @@ class NemotronHForCausalLM(
|
||||
Tuple containing:
|
||||
- conv_state_shape: Shape for convolutional state cache
|
||||
- temporal_state_shape: Shape for state space model cache
|
||||
- (when use_replayssm is enabled) the x_cache/dt_cache/B_cache
|
||||
ring-buffer shapes
|
||||
- x_cache/dt_cache/B_cache ring-buffer shapes (use_replayssm only)
|
||||
"""
|
||||
parallel_config = vllm_config.parallel_config
|
||||
cache_config = vllm_config.cache_config
|
||||
|
||||
Reference in New Issue
Block a user