Fix a typo in error message (#227)

By the code context, it can be inferred that do_sample==False when control reaches this point.
pull/225/head^2
Shuchang Zhou 1 year ago committed by GitHub
parent fa5ac6e3b4
commit 3189b395f0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -129,7 +129,7 @@ class RemoteGenerationMixin:
decoding_algorithm = BeamSearchAlgorithm(num_beams, batch_size=batch_size)
else:
if top_k is not None or top_p is not None:
logger.warning("You passed top_k or top_p but did pass do_sample=True. Running greedy sampling")
logger.warning("You passed top_k or top_p but did not pass do_sample=True. Running greedy sampling")
decoding_algorithm = GreedyAlgorithm()
if num_beams > 1:

Loading…
Cancel
Save