add out_seq_length for api prompt

53dc7e6e · zihanl · e755f6fd · 53dc7e6e
Commit 53dc7e6e authored Dec 09, 2021 by zihanl
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tasks/knwl_dialo/prompt.py tasks/knwl_dialo/prompt.py +1 -1

No files found.
--- a/tasks/knwl_dialo/prompt.py
+++ b/tasks/knwl_dialo/prompt.py
@@ -144,7 +144,7 @@ def generate_samples_by_calling_api():
        # get the output generations from the api, 
        # and write to the output file
-        generations = call_model_api(inputs)
+        generations = call_model_api(inputs, args.out_seq_length)
        fname_out.write(generations)
        fname_out.write("\n")