We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1c8a702 commit f02f59bCopy full SHA for f02f59b
tests/ut/ops/test_vocab_parallel_embedding.py
@@ -227,5 +227,5 @@ def test_get_logits(self):
227
lmhead.quant_method = self.mock_quant_method
228
lmhead.quant_method.apply = self.mock_quant_method.apply
229
hidden_state = torch.randn(1, self.org_num_embeddings)
230
- processed_logits = processor._get_logits(hidden_state, lmhead)
+ processor._get_logits(hidden_state, lmhead)
231
self.mock_quant_method.apply.assert_called_once()
0 commit comments