Skip to content

GitLab

Unverified Commit a105c9b7 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

fix (#13383)

parent 4475f1dc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_speech_encoder_decoder.py
View file @ a105c9b7
......@@ -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,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment