transformers
ef0f85cd - [Vision] `.to` function for ImageProcessors (#20536)
Commit
View On GitHub
Commit
3 years ago
[Vision] `.to` function for ImageProcessors (#20536) * add v1 with tests * add checker * simplified version * update docstring * better version * fix docstring + change order * make style * tests + change conditions * final tests * modify docstring * Update src/transformers/feature_extraction_utils.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * replace by `ValueError` * fix logic * apply suggestions * `dtype` is not needed * adapt suggestions * remove `_parse_args_to_device` Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Author
younesbelkada younesbelkada
Parents
67d32f46
FAQ Terms Privacy Refunds Impressum
Loading