move topk capturers to srt/state_capturer/ (#24450)
Co-authored-by: Yueming Yuan <yym022502@gmail.com> Co-authored-by: fzyzcjy <5236035+fzyzcjy@users.noreply.github.com> Co-authored-by: Ziang Li <ziangli@umich.edu>
This commit is contained in:
co-authored by
Yueming Yuan
fzyzcjy
Ziang Li
parent
47a416fc62
commit
08d4c2072b
@@ -9,7 +9,7 @@ import torch
|
||||
from torch.nn.utils.rnn import pad_sequence
|
||||
|
||||
from sglang.benchmark.utils import download_and_cache_hf_file
|
||||
from sglang.srt.layers.moe.routed_experts_capturer import (
|
||||
from sglang.srt.state_capturer.routed_experts import (
|
||||
extract_routed_experts_from_meta_info,
|
||||
)
|
||||
from sglang.srt.utils import kill_process_tree
|
||||
|
||||
Reference in New Issue
Block a user