mirror of
https://github.com/deepseek-ai/DeepSeek-Coder
synced 2024-12-04 18:14:44 +00:00
commit
db4ada222f
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user