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

fix typos in the tests directory (#36932)

Browse Source 
* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: fix typos in test codes

* chore: format codes
This commit is contained in:
Afanti
2025-03-25 17:49:24 +08:00
committed by GitHub
parent 69bcb86c58
commit 19085c28da
9 changed files with 19 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/mimi/test_modeling_mimi.py
View File

@@ -485,7 +485,7 @@ class MimiIntegrationTest(unittest.TestCase):
for num_codebooks, expected_rmse in expected_rmse.items():
with torch.no_grad():
# use max bandwith for best possible reconstruction
# use max bandwidth for best possible reconstruction
encoder_outputs = model.encode(inputs["input_values"], num_quantizers=int(num_codebooks))
audio_codes = encoder_outputs[0]
@@ -537,7 +537,7 @@ class MimiIntegrationTest(unittest.TestCase):
model = MimiModel.from_pretrained(model_id, use_cache=use_cache).to(torch_device)
for num_codebooks, expected_rmse in expected_rmses.items():
with torch.no_grad():
# use max bandwith for best possible reconstruction
# use max bandwidth for best possible reconstruction
encoder_outputs = model.encode(inputs["input_values"], num_quantizers=int(num_codebooks))
audio_code_sums = encoder_outputs[0].sum().cpu().item()