Commit 53dc7e6e authored by zihanl's avatar zihanl
Browse files

add out_seq_length for api prompt

parent e755f6fd

tasks/knwl_dialo/prompt.py

+1 −1
Original line number Diff line number Diff line
@@ -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")