SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix bad tensor shape in failing Hubert test. (#39502)

Browse Source 
Fix bad tensor shape in Hubert test.
This commit is contained in:
Eric Bezzam
2025-07-21 13:25:52 +02:00
committed by GitHub
parent 39ba5f3cc2
commit 78fb2d2760
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/models/hubert/test_modeling_hubert.py
View File

@@ -962,19 +962,23 @@ class HubertModelIntegrationTest(unittest.TestCase):
# model(wav)['dense'] # model(wav)['dense']
expected_outputs_first = torch.tensor( expected_outputs_first = torch.tensor(
[ [
[0.0267, 0.1776, -0.1706, -0.4559], [
[-0.2430, -0.2943, -0.1864, -0.1187], [0.0267, 0.1776, -0.1706, -0.4559],
[-0.1812, -0.4239, -0.1916, -0.0858], [-0.2430, -0.2943, -0.1864, -0.1187],
[-0.1495, -0.4758, -0.4036, 0.0302], [-0.1812, -0.4239, -0.1916, -0.0858],
[-0.1495, -0.4758, -0.4036, 0.0302],
]
], ],
device=torch_device, device=torch_device,
) )
expected_outputs_last = torch.tensor( expected_outputs_last = torch.tensor(
[ [
[0.3366, -0.2734, -0.1415, -0.3055], [
[0.2329, -0.3580, -0.1421, -0.3197], [0.3366, -0.2734, -0.1415, -0.3055],
[0.1631, -0.4301, -0.1965, -0.2956], [0.2329, -0.3580, -0.1421, -0.3197],
[0.3342, -0.2185, -0.2253, -0.2363], [0.1631, -0.4301, -0.1965, -0.2956],
[0.3342, -0.2185, -0.2253, -0.2363],
]
], ],
device=torch_device, device=torch_device,
) )