diff --git a/self_instruct/bootstrap_instructions.py b/self_instruct/bootstrap_instructions.py index 3ced1b37..e2d2da82 100644 --- a/self_instruct/bootstrap_instructions.py +++ b/self_instruct/bootstrap_instructions.py @@ -171,7 +171,7 @@ def parse_args(): prompt = encode_prompt(prompt_instructions, classification=args.use_clf_seed_tasks_only) batch_inputs.append(prompt) results = make_gpt3_requests( - engine="davinci", + engine=args.engine, prompts=batch_inputs, max_tokens=1024, temperature=0.7,