Skip to content

Conversation

@lkm2835
Copy link
Contributor

@lkm2835 lkm2835 commented Nov 20, 2024

For tensor parallelism, sharing the embedding table must setuse_parallel_embedding=True and embedding_sharding_dim=0
-> For tensor parallelism, sharing the embedding table must set use_parallel_embedding=True and embedding_sharding_dim=0

@hello-11 hello-11 self-assigned this Nov 25, 2024
@hello-11 hello-11 added the triaged Issue has been triaged by maintainers label Nov 25, 2024
Copy link
Member

@kaiyux kaiyux left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks very much for fixing it!

@kaiyux kaiyux merged commit 4420547 into NVIDIA:main Dec 2, 2024
wu1du2 pushed a commit to wu1du2/TensorRT-LLM that referenced this pull request May 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

triaged Issue has been triaged by maintainers

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants