PR #28054 Make GPT2 traceable in meta state

Make GPT2 traceable in meta state #28054

younesbelkada merged 2 commits into huggingface:main from kwen2501:gpt2_meta_trace

Put device in tensor constructor instead of to()

c520fd1b

younesbelkada commented on 2023-12-15

muellerzr approved these changes on 2023-12-15

SunMarc requested a review from

SunMarc 2 years ago

SunMarc approved these changes on 2023-12-15

Fix copy

d9bc8fd0

younesbelkada approved these changes on 2023-12-15

younesbelkada requested a review from

amyeroberts 2 years ago

amyeroberts approved these changes on 2023-12-15

younesbelkada merged 74cae670 into main 2 years ago

Reviewers

SunMarc

amyeroberts

younesbelkada

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone