PR #777 fix transformers tests

fix transformers tests #777

ArthurZucker merged 2 commits into huggingface:main from ArthurZucker:fix_get_balanced_memory

fix range of the loop, last device was not leftout

545ccea2

younesbelkada commented on 2022-10-19

Update based on review ,

6787006e

ArthurZucker requested a review from

sgugger 3 years ago

sgugger approved these changes on 2022-10-19

ArthurZucker merged 87a7e078 into main 3 years ago

Reviewers

sgugger

younesbelkada

Assignees

No one assigned

Labels

None yet

Milestone

No milestone