fix (#13383)

2021-09-01 23:12:01 +02:00
parent 4475f1dc2a
commit a105c9b776
1 changed files with 1 additions and 0 deletions
--- 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,