PR #48782 Add tensor parallel support for GPT-Neo

Commits

Key the tied-embedding TP rule on the model's input embedding attribute

Siddhant231xyz committed 10 days ago
Add tensor parallel support for GPT-Neo

Siddhant231xyz committed 10 days ago

transformers Add tensor parallel support for GPT-Neo #48782 Open