riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity
4cd7ed4b3b
transformers/tests/vision_encoder_decoder
History
Yih-Dar 4cd7ed4b3b
Fix a TF Vision Encoder Decoder test (#15896) 
* send PyTorch inputs to the correct device

* Fix: TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-03-03 13:21:31 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_flax_vision_encoder_decoder.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_tf_vision_encoder_decoder.py Fix a TF Vision Encoder Decoder test (#15896) 2022-03-03 13:21:31 +01:00
test_modeling_vision_encoder_decoder.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00