Fix typo (#2473)

lkm2835 · web-flow · commit 4420547017fd · 2024-12-02T10:11:27.000+08:00
diff --git a/tensorrt_llm/models/modeling_utils.py b/tensorrt_llm/models/modeling_utils.py
@@ -362,7 +362,7 @@ def __init__(self,
             if (not use_parallel_embedding) or (use_parallel_embedding and
                                                 embedding_sharding_dim == 1):
                 raise NotImplementedError(
-                    "For tensor parallelism, sharing the embedding table must set" \
+                    "For tensor parallelism, sharing the embedding table must set " \
                         "use_parallel_embedding=True and embedding_sharding_dim=0"
                 )
         if share_embedding_table and mapping.pp_size > 1:

Original file line number	Diff line number	Diff line change
`@@ -362,7 +362,7 @@ def __init__(self,`
`362`	`362`	`if (not use_parallel_embedding) or (use_parallel_embedding and`
`363`	`363`	`embedding_sharding_dim == 1):`
`364`	`364`	`raise NotImplementedError(`
`365`		`- "For tensor parallelism, sharing the embedding table must set" \`
	`365`	`+ "For tensor parallelism, sharing the embedding table must set " \`
`366`	`366`	`"use_parallel_embedding=True and embedding_sharding_dim=0"`
`367`	`367`	`)`
`368`	`368`	`if share_embedding_table and mapping.pp_size > 1:`