[fix] Fixes accuracy issues caused by incorrect use of rope (#13495)

This commit is contained in:
Baidu-AIAK
2025-11-18 14:17:33 -08:00
committed by GitHub
parent c0d1a3383b
commit 9b64f6f359
@@ -168,7 +168,7 @@ class Indexer(CustomOp):
max_position=max_position_embeddings, max_position=max_position_embeddings,
base=rope_theta, # type: ignore base=rope_theta, # type: ignore
rope_scaling=rope_scaling, rope_scaling=rope_scaling,
is_neox_style=False, is_neox_style=True,
device=get_global_server_args().device, device=get_global_server_args().device,
) )
self.block_size = block_size self.block_size = block_size