diff --git a/Evaluation/HumanEval/eval_instruct.py b/Evaluation/HumanEval/eval_instruct.py index 71874fe..34bea56 100644 --- a/Evaluation/HumanEval/eval_instruct.py +++ b/Evaluation/HumanEval/eval_instruct.py @@ -67,7 +67,7 @@ def generate_main(args): generated_examples = [] for ex in tqdm(examples, desc='Generating'): - gen_example = generate_one(ex, lang, tokenizer, model) + gen_example = generate_one(ex, args.language, tokenizer, model) generated_examples.append(gen_example) print("Generate all over!!!") @@ -125,4 +125,4 @@ if __name__ == '__main__': os.environ["TOKENIZERS_PARALLELISM"] = "false" generate_main(args) - pass \ No newline at end of file + pass