transformers
44127ec6 - Fix test for auto_find_batch_size on multi-GPU (#27947)

Commit

2 years ago

Fix test for auto_find_batch_size on multi-GPU (#27947) * Fix test for multi-GPU * WIth CPU handle

References

Author

muellerzr

Parents