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

fix kyutai tests (#39416)

Browse Source 
* fix

* fix

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Co-authored-by: eustlb <94853470+eustlb@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2025-07-25 18:42:04 +02:00
committed by GitHub
parent 4f17bf0572
commit 4b125e2993
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/models/kyutai_speech_to_text/test_modeling_kyutai_speech_to_text.py
View File

@@ -777,4 +777,11 @@ class KyutaiSpeechToTextForConditionalGenerationIntegrationTests(unittest.TestCa
]) ])
# fmt: on # fmt: on
torch.testing.assert_close(out.cpu(), EXPECTED_TOKENS) # See https://github.com/huggingface/transformers/pull/39416
EXPECTED_TOKENS_2 = torch.clone(EXPECTED_TOKENS)
EXPECTED_TOKENS_2[2, 159:162] = torch.tensor([3, 0, 269])
try:
torch.testing.assert_close(out.cpu(), EXPECTED_TOKENS)
except AssertionError:
torch.testing.assert_close(out.cpu(), EXPECTED_TOKENS_2)