Browse Source

Fix a typo in error message (#227)

By the code context, it can be inferred that do_sample==False when control reaches this point.
Shuchang Zhou 2 years ago
parent
commit
3189b395f0
1 changed files with 1 additions and 1 deletions
  1. 1 1
      src/petals/client/remote_generation.py

+ 1 - 1
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: