From 3867c6431ae3388eda30d3f3f960d32abe380273 Mon Sep 17 00:00:00 2001 From: jianzhao-xu <978716854@qq.com> Date: Fri, 27 Mar 2026 02:23:30 +0800 Subject: [PATCH] Fix bug in dbrx model (#21445) Co-authored-by: Jianzhao Xu --- python/sglang/srt/models/dbrx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/models/dbrx.py b/python/sglang/srt/models/dbrx.py index e37827565..0c7e16052 100644 --- a/python/sglang/srt/models/dbrx.py +++ b/python/sglang/srt/models/dbrx.py @@ -211,7 +211,7 @@ class DbrxAttention(nn.Module): self.head_dim = self.d_model // self.total_num_heads self.total_num_kv_heads = config.attn_config.kv_n_heads self.clip_qkv = config.attn_config.clip_qkv - self.rope_theta = config.attn_config.rope_parameters["rope_theta"] + self.rope_theta = config.attn_config.rope_theta self.max_position = config.max_seq_len # pylint: disable=invalid-name