transformers
Add tensor parallel support for GPT-Neo
#48782
Open
Overview
Commits
2
Changes
View On GitHub

Add tensor parallel support for GPT-Neo #48782

Siddhant231xyz wants to merge 2 commits into huggingface:main from Siddhant231xyz:tp-gpt-neo
Siddhant231xyz
Siddhant231xyz Key the tied-embedding TP rule on the model's input embedding attribute
10435529
Siddhant231xyz Add tensor parallel support for GPT-Neo
fe59c1a2
github-actions
Siddhant231xyz Siddhant231xyz force pushed from 79678304 to fe59c1a2 3 days ago
github-actions

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum