Update eval_instruct.py

This commit is contained in:
Talmeez Fuaad 2023-11-26 17:11:21 +05:00 committed by GitHub
parent 0504740537
commit 020e786bb8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

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