transformers
Fix test for auto_find_batch_size on multi-GPU
#27947
Merged

Fix test for auto_find_batch_size on multi-GPU #27947

muellerzr merged 2 commits into main from fix-multigpu-tests-some
muellerzr
muellerzr Fix test for multi-GPU
46eaa635
muellerzr WIth CPU handle
24ef7c56
muellerzr muellerzr requested a review from amyeroberts amyeroberts 2 years ago
amyeroberts
amyeroberts approved these changes on 2023-12-11
muellerzr muellerzr merged 44127ec6 into main 2 years ago
muellerzr muellerzr deleted the fix-multigpu-tests-some branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone