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

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