From 3189b395f0031daaaf529d571773dc9fb8200205 Mon Sep 17 00:00:00 2001 From: Shuchang Zhou Date: Thu, 19 Jan 2023 22:38:43 +0800 Subject: [PATCH] Fix a typo in error message (#227) By the code context, it can be inferred that do_sample==False when control reaches this point. --- src/petals/client/remote_generation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: