transformers
HF <-> megatron checkpoint reshaping and conversion for GPT
#19317

Merged

HF <-> megatron checkpoint reshaping and conversion for GPT #19317

pacman100 merged 4 commits into main from smangrul/megatron-hf-conversion-utils

HF <-> megatron checkpoint conversion handling reshaping from differe…

b5773bd6

pacman100 requested a review from

sgugger 3 years ago

sgugger commented on 2022-10-04

Apply suggestions from code review

dc553a75

addressing comments

2091e0f9

sgugger approved these changes on 2022-10-07

add doc strings and 🐛 fixes

28850fb3

pacman100 merged 56af8df3 into main 3 years ago

pacman100 deleted the smangrul/megatron-hf-conversion-utils branch 3 years ago

Reviewers

sgugger

Assignees

No one assigned

Labels

None yet

Milestone

No milestone