transformers
Add tensor parallel support for GPT-J
#48783
Open

Add tensor parallel support for GPT-J #48783

Siddhant231xyz wants to merge 2 commits into huggingface:main from Siddhant231xyz:tp-gptj
Siddhant231xyz
Siddhant231xyz Key the tied-embedding TP rule on the model's input embedding attribute
10435529
Siddhant231xyz Add tensor parallel support for GPT-J
41ef36e4
github-actions
Siddhant231xyz Siddhant231xyz force pushed from c4cd5f25 to 41ef36e4 5 days ago
github-actions

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone