[3/n] deepseek_v2.py Refactor: Migrate MLA forward method in deepseek_v2.py (#19122)

This commit is contained in:
Baizhou Zhang
2026-02-27 13:37:29 -08:00
committed by GitHub
parent 7e46aafebb
commit 776709efe8
9 changed files with 906 additions and 818 deletions
+2 -2
View File
@@ -3,9 +3,9 @@ import unittest
import torch
from utils import precision
from sglang.srt.layers.rotary_embedding import (
from sglang.srt.layers.rotary_embedding.base import RotaryEmbedding
from sglang.srt.layers.rotary_embedding.rope_variant import (
DeepseekScalingRotaryEmbedding,
RotaryEmbedding,
)
from sglang.srt.server_args import ServerArgs, set_global_server_args_for_scheduler
from sglang.test.test_utils import CustomTestCase