Artem Chumachenko 2 年之前
父節點
當前提交
5ef7fea883
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      src/client/remote_generation.py

+ 1 - 0
src/client/remote_generation.py

@@ -61,6 +61,7 @@ class RemoteGenerationMixin:
         :param decoding_algorithm: The decoding algorithm to use.
         :param decoding_algorithm: The decoding algorithm to use.
         :param provided_constraints: A list of constraints to use.
         :param provided_constraints: A list of constraints to use.
         :param model_kwargs: Additional arguments to pass to the model.
         :param model_kwargs: Additional arguments to pass to the model.
+        :param num_return_sequences: How many hypothesis from the beam will be in output.
         """
         """
 
 
         assert (
         assert (