From 564f279f7a35b91d1a6beb5e3f73179eedc95b72 Mon Sep 17 00:00:00 2001 From: lilujia Date: Wed, 29 Oct 2025 11:28:42 +0800 Subject: [PATCH] [XPU] fix pos_emb_type bug --- fastdeploy/worker/xpu_model_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastdeploy/worker/xpu_model_runner.py b/fastdeploy/worker/xpu_model_runner.py index 3a480c8df60..f88a1147792 100644 --- a/fastdeploy/worker/xpu_model_runner.py +++ b/fastdeploy/worker/xpu_model_runner.py @@ -850,7 +850,7 @@ def _init_share_inputs(self, max_num_seqs: int): else: # neox style = False rope_head_dim = head_dim // 2 - if head_dim == self.model_config.head_dim: + if rope_head_dim == self.model_config.head_dim: self.share_inputs["pos_emb_type"] = "NORMAL" else: self.share_inputs["pos_emb_type"] = "HALF_HEAD_DIM"