From 935a76d90d43be58b62efce28e713c59d21b2191 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 28 Feb 2022 11:23:13 +0100 Subject: [PATCH] [UniSpeechSat] correct unispeech sat (#15847) --- tests/unispeech_sat/test_modeling_unispeech_sat.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unispeech_sat/test_modeling_unispeech_sat.py b/tests/unispeech_sat/test_modeling_unispeech_sat.py index bae49c7b4c..da4359659a 100644 --- a/tests/unispeech_sat/test_modeling_unispeech_sat.py +++ b/tests/unispeech_sat/test_modeling_unispeech_sat.py @@ -857,10 +857,10 @@ class UniSpeechSatModelIntegrationTest(unittest.TestCase): # fmt: off expected_hidden_states_slice = torch.tensor( - [[[-0.1192, -0.0825], - [-0.0012, 0.0235]], - [[-0.1240, -0.1332], - [-0.0658, -0.0565]]], + [[[-0.1172, -0.0797], + [-0.0012, 0.0213]], + [[-0.1225, -0.1277], + [-0.0668, -0.0585]]], device=torch_device, ) # fmt: on