Преглед изворни кода

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 година
родитељ
комит
3189b395f0
1 измењених фајлова са 1 додато и 1 уклоњено
  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)
                 decoding_algorithm = BeamSearchAlgorithm(num_beams, batch_size=batch_size)
             else:
             else:
                 if top_k is not None or top_p is not None:
                 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()
                 decoding_algorithm = GreedyAlgorithm()
 
 
         if num_beams > 1:
         if num_beams > 1: