update pre-commit config (#18860)
This commit is contained in:
@@ -376,7 +376,7 @@ class LogitsProcessor(nn.Module):
|
||||
|
||||
logprobs_result = self.process_input_logprobs(input_logits, logits_metadata)
|
||||
else:
|
||||
(logprobs_result, sampled_logits) = self.process_input_logprobs_by_chunk(
|
||||
logprobs_result, sampled_logits = self.process_input_logprobs_by_chunk(
|
||||
pruned_states,
|
||||
sample_indices,
|
||||
input_logprob_indices,
|
||||
|
||||
Reference in New Issue
Block a user