Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
dea8aa7a
"vscode:/vscode.git/clone" did not exist on "976970554d95d96b802faa32488d4bf836678340"
Unverified
Commit
dea8aa7a
authored
Jun 20, 2025
by
Li Hui
Committed by
GitHub
Jun 19, 2025
Browse files
Clean unused import for mimo mtp model (#7370)
parent
906dbc34
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
18 deletions
+2
-18
python/sglang/srt/models/mimo_mtp.py
python/sglang/srt/models/mimo_mtp.py
+2
-18
No files found.
python/sglang/srt/models/mimo_mtp.py
View file @
dea8aa7a
...
@@ -7,33 +7,17 @@ import torch
...
@@ -7,33 +7,17 @@ import torch
from
torch
import
nn
from
torch
import
nn
from
transformers
import
PretrainedConfig
from
transformers
import
PretrainedConfig
from
sglang.srt.distributed
import
(
from
sglang.srt.distributed
import
get_tensor_model_parallel_world_size
get_tensor_model_parallel_rank
,
get_tensor_model_parallel_world_size
,
split_tensor_along_last_dim
,
tensor_model_parallel_all_gather
,
)
from
sglang.srt.layers.layernorm
import
RMSNorm
from
sglang.srt.layers.layernorm
import
RMSNorm
from
sglang.srt.layers.linear
import
QKVParallelLinear
,
RowParallelLinear
from
sglang.srt.layers.logits_processor
import
LogitsProcessor
from
sglang.srt.layers.logits_processor
import
LogitsProcessor
from
sglang.srt.layers.pooler
import
Pooler
,
PoolingType
from
sglang.srt.layers.quantization.base_config
import
QuantizationConfig
from
sglang.srt.layers.quantization.base_config
import
QuantizationConfig
from
sglang.srt.layers.radix_attention
import
RadixAttention
from
sglang.srt.layers.rotary_embedding
import
get_rope
from
sglang.srt.layers.vocab_parallel_embedding
import
(
from
sglang.srt.layers.vocab_parallel_embedding
import
(
ParallelLMHead
,
ParallelLMHead
,
VocabParallelEmbedding
,
VocabParallelEmbedding
,
)
)
from
sglang.srt.model_executor.forward_batch_info
import
ForwardBatch
from
sglang.srt.model_executor.forward_batch_info
import
ForwardBatch
from
sglang.srt.model_loader.weight_utils
import
default_weight_loader
from
sglang.srt.model_loader.weight_utils
import
default_weight_loader
from
sglang.srt.models.mimo
import
MiMoForCausalLM
from
sglang.srt.models.qwen2
import
Qwen2DecoderLayer
from
sglang.srt.models.qwen2
import
(
Qwen2Attention
,
Qwen2DecoderLayer
,
Qwen2MLP
,
Qwen2Model
,
)
from
sglang.srt.utils
import
add_prefix
class
MiMoMultiTokenPredictorLayer
(
nn
.
Module
):
class
MiMoMultiTokenPredictorLayer
(
nn
.
Module
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment