From 607611f2408ecba5f96bf72ed82acf8713758c12 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 6 Sep 2021 16:09:24 +0200 Subject: [PATCH] up (#13448) --- tests/test_modeling_encoder_decoder.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_modeling_encoder_decoder.py b/tests/test_modeling_encoder_decoder.py index 5f7dce94df..478d49dde0 100644 --- a/tests/test_modeling_encoder_decoder.py +++ b/tests/test_modeling_encoder_decoder.py @@ -201,6 +201,7 @@ class EncoderDecoderMixin: with tempfile.TemporaryDirectory() as tmpdirname: enc_dec_model.save_pretrained(tmpdirname) enc_dec_model = EncoderDecoderModel.from_pretrained(tmpdirname) + enc_dec_model.to(torch_device) after_outputs = enc_dec_model( input_ids=input_ids, @@ -245,6 +246,7 @@ class EncoderDecoderMixin: encoder_pretrained_model_name_or_path=encoder_tmp_dirname, decoder_pretrained_model_name_or_path=decoder_tmp_dirname, ) + enc_dec_model.to(torch_device) after_outputs = enc_dec_model( input_ids=input_ids,