Merge pull request #52 from itstalmeez/patch-1

Update eval_instruct.py
This commit is contained in:
DejianYang
2023-11-27 13:59:23 +08:00
committed by GitHub

View File

@@ -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
pass