[VocabParallelEmbedding] fix extra_repr fields concat (#51224)
Signed-off-by: Andy Xie <andy.xning@gmail.com>
Ning Xie committed
Aug 6, 2026 at 20:22 UTC
62a86318de3655f970baf7c2ff89c81a72c1a1b3
1 file changed
+2
-1
vllm/model_executor/layers/vocab_parallel_embedding.py
+2
-1
@@ -507,7 +507,8 @@ class VocabParallelEmbedding(PluggableLayer):
507
return output_parallel
508
509
def extra_repr(self) -> str:
510
- s = f"num_embeddings={self.num_embeddings_per_partition}"
510
+ s = f"num_embeddings={self.num_embeddings}"
511
+ s += f", num_embeddings_per_partition={self.num_embeddings_per_partition}"
512
s += f", embedding_dim={self.embedding_dim}"
513
s += f", org_vocab_size={self.org_vocab_size}"
514
s += f", num_embeddings_padded={self.num_embeddings_padded}"