From a105c9b77677dbab44b0f3e50981292dd11d3fad Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 1 Sep 2021 23:12:01 +0200 Subject: [PATCH] fix (#13383) --- tests/test_modeling_speech_encoder_decoder.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_speech_encoder_decoder.py b/tests/test_modeling_speech_encoder_decoder.py index 783d6b7ec0..5c42f6ec8c 100644 --- a/tests/test_modeling_speech_encoder_decoder.py +++ b/tests/test_modeling_speech_encoder_decoder.py @@ -184,6 +184,7 @@ class EncoderDecoderMixin: with tempfile.TemporaryDirectory() as tmpdirname: enc_dec_model.save_pretrained(tmpdirname) enc_dec_model = SpeechEncoderDecoderModel.from_pretrained(tmpdirname) + enc_dec_model.to(torch_device) after_outputs = enc_dec_model( input_values=input_values,