diff --git a/src/petals/client/remote_generation.py b/src/petals/client/remote_generation.py index af4166d..4ea0c9b 100644 --- a/src/petals/client/remote_generation.py +++ b/src/petals/client/remote_generation.py @@ -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: