Skip to content

Commit 9956351

Browse files
committed
linting
Your commit message here Co-authored-by: Codex Signed-off-by: Yoav Miron <yomiron@nvidia.com>
1 parent 36a76de commit 9956351

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

vllm/config/speculative.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,7 @@ def compute_hash(self) -> str:
611611
"extract_hidden_states",
612612
"dflash",
613613
"dspark",
614-
"mtp"
614+
"mtp",
615615
)
616616
factors.append(uses_aux_hidden_states)
617617

vllm/model_executor/models/nemotron_h_mtp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -446,7 +446,8 @@ def load_weights(self, weights: Iterable[tuple[str, torch.Tensor]]) -> set[str]:
446446
is_lm_head_weight = name.startswith("lm_head.")
447447
if is_lm_head_weight:
448448
self.has_own_lm_head = True
449-
# Only process MTP and LM head weights - skip all non-MTP and non-LM head weights
449+
# Only process MTP and LM head weights -
450+
# skip all non-MTP and non-LM head weights
450451
if (
451452
not name.startswith("mtp.")
452453
and "embeddings" not in name

0 commit comments

Comments
 (0)