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
commit db4ada222f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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