Skip to content

Commit f02f59b

Browse files
committed
rebase main and fix
Signed-off-by: zzhx1 <zzh_201018@outlook.com>
1 parent 1c8a702 commit f02f59b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/ut/ops/test_vocab_parallel_embedding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -227,5 +227,5 @@ def test_get_logits(self):
227227
lmhead.quant_method = self.mock_quant_method
228228
lmhead.quant_method.apply = self.mock_quant_method.apply
229229
hidden_state = torch.randn(1, self.org_num_embeddings)
230-
processed_logits = processor._get_logits(hidden_state, lmhead)
230+
processor._get_logits(hidden_state, lmhead)
231231
self.mock_quant_method.apply.assert_called_once()

0 commit comments

Comments
 (0)