DeepSeek/DeepSeek-Coder
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Coder synced 2025-06-26 18:25:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix add_generation_prompt in latest version

Browse Source
This commit is contained in:
Dejian Yang
2024-01-09 21:55:21 +08:00
parent d3bb741e25
commit c160d96860
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Evaluation/HumanEval/eval_instruct.py
View File

@@ -39,7 +39,7 @@ def generate_one(example, lang, tokenizer, model):
eos_token_id=stop_id
)
output = tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True)
output = tokenizer.decode(outputs[0][len(inputs[0]):], skip_special_tokens=True, add_generation_prompt=True)
example['output'] = output
return extract_generation_code(example, lang_code=lang)