fix syntax error

Signed-off-by: Gerald Shen <[email protected]>
NVIDIA · Jul 12, 2023 · acbfe08 · acbfe08
1 parent b06588e
commit acbfe08
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/nlp/modules/common/text_generation_server.py b/nemo/collections/nlp/modules/common/text_generation_server.py
@@ -140,7 +140,7 @@ def put(self):
                 return "repetition_penalty must be a positive number no less than 1.0"
             if not (1.0 <= repetition_penalty):
                 return "repetition_penalty must be a positive number no less than 1.0"
-        
+
         end_strings = ['<|endoftext|>']
         if 'end_strings' in request.get_json():
             end_strings = request.get_json()['end_strings']
@@ -191,7 +191,7 @@ def put(self):
                 greedy,
                 repetition_penalty,
                 end_strings=end_strings,
-                min_tokens_to_generate,
+                min_tokens_to_generate=min_tokens_to_generate,
                 **extra,
             )
             for k in output: