Skip to content

Commit

Permalink
Fixes typo in "prefilling"
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 577882571
  • Loading branch information
mlbileschi authored and t5-copybara committed Oct 30, 2023
1 parent 55b32b5 commit 397236e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions t5x/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -675,13 +675,13 @@ def predict_batch_with_aux(
# Prepare autoregressive cache.
if 'initial_index' not in inspect.signature(self.decode_fn).parameters:
logging.info(
'Disabling prompt prefilliing due to incompatible decode fn: %s.',
'Disabling prompt prefilling due to incompatible decode fn: %s.',
self.decode_fn,
)
prefill_decoder_prompt = False
elif 'prefill' not in inspect.signature(self.module.decode).parameters:
logging.info(
'Disabling prompt prefilliing due to incompatible `module.decode`.'
'Disabling prompt prefilling due to incompatible `module.decode`.'
)
prefill_decoder_prompt = False
else:
Expand Down

0 comments on commit 397236e

Please sign in to comment.