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

Fix typos in tests (#36547)

Browse Source 
Signed-off-by: co63oc <co63oc@users.noreply.github.com>
This commit is contained in:
co63oc
2025-03-06 07:04:06 +08:00
committed by GitHub
parent 752ef3fd4e
commit 996f512d52
99 changed files with 282 additions and 282 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/mllama/test_modeling_mllama.py
View File

@@ -392,7 +392,7 @@ class MllamaForConditionalGenerationModelTest(ModelTesterMixin, GenerationTester
pass
@pytest.mark.generate
# overriden because mllama has special cache for self and cross attentions
# overridden because mllama has special cache for self and cross attentions
def test_past_key_values_format(self):
# Test that the KV cache is formatted correctly. Exceptions need to explicitly overwrite this test. Having a
# standard KV cache format is important for a consistent API (and for advanced generation methods).
@@ -444,7 +444,7 @@ class MllamaForConditionalGenerationModelTest(ModelTesterMixin, GenerationTester
past_kv[i][1].shape, (batch_size, num_attention_heads, seq_length, per_head_embed_dim)
)
# overriden because mllama has special cache for self and cross attentions
# overridden because mllama has special cache for self and cross attentions
def _check_past_key_values_for_generate(self, batch_size, decoder_past_key_values, cache_length, config):
self.assertIsInstance(decoder_past_key_values, Cache)
self.assertListEqual(