transformers
When save a model on TPU, make a copy to be moved to CPU
#27993

Merged

When save a model on TPU, make a copy to be moved to CPU #27993

amyeroberts merged 3 commits into huggingface:main from qihqi:fix_save_tpu

When save a model, make a copy to be moved to CPU, dont move the orig…

c658ffe6

qihqi force pushed to c658ffe6 2 years ago

make deepcopy inside of _save_tpu

3cbee0f6

qihqi force pushed to 3cbee0f6 2 years ago

muellerzr approved these changes on 2023-12-15

muellerzr requested a review from

ArthurZucker 2 years ago

amyeroberts removed review request from

ArthurZucker 2 years ago

amyeroberts requested a review from

amyeroberts 2 years ago

amyeroberts requested a review from

ArthurZucker 2 years ago

amyeroberts commented on 2023-12-15

Move to tpu without copy

13f47747

muellerzr requested a review from

amyeroberts 2 years ago

muellerzr removed review request from

ArthurZucker 2 years ago

amyeroberts approved these changes on 2023-12-18

amyeroberts merged 5aec50ec into main 2 years ago

Reviewers

amyeroberts

muellerzr

Assignees

No one assigned

Labels

None yet

Milestone

No milestone