diff --git a/python/sglang/srt/server_args.py b/python/sglang/srt/server_args.py index ac774b82a336..2658b29b23e7 100644 --- a/python/sglang/srt/server_args.py +++ b/python/sglang/srt/server_args.py @@ -4264,8 +4264,8 @@ def _validate_mamba_extra_buffer(self, view, model_arch: str): view, model_arch ), f"extra_buffer is not supported for {model_arch}; use no_buffer." assert ( - is_cuda() or is_musa() or is_npu() - ), "extra_buffer needs CUDA/MUSA/NPU (FLA)." + is_cuda() or is_musa() or is_npu() or is_hip() + ), "extra_buffer needs CUDA/MUSA/NPU/ROCm (FLA)." if view.speculative_num_draft_tokens is not None: assert ( view.mamba_radix_cache_strategy != "extra_buffer_lazy"