Skip to content

Commit 4edc57e

Browse files
committed
rm sampler
Signed-off-by: wangli <wangli858794774@gmail.com>
1 parent 2972684 commit 4edc57e

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

vllm_ascend/models/deepseek_mtp.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,6 @@
3636
from vllm.model_executor.models.deepseek_v2 import DeepseekV2DecoderLayer
3737
from vllm.model_executor.models.utils import maybe_prefix
3838
from vllm.sequence import IntermediateTensors
39-
from vllm.v1.sample.sampler import Sampler
4039

4140

4241
class CustomDeepSeekShareHead(SharedHead):
@@ -186,7 +185,6 @@ def __init__(self, *, vllm_config: VllmConfig, prefix: str = ""):
186185
self.model = CustomDeepSeekMultiTokenPredictor(vllm_config=vllm_config,
187186
prefix=maybe_prefix(
188187
prefix, "model"))
189-
self.sampler = Sampler()
190188

191189
def forward(
192190
self,

0 commit comments

Comments
 (0)