github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #857 from wbbeyourself/fix-ctrl-c-exception 

fix: eval_llm.py Ctrl+C 退出异常修复
pull/860/head
jingyaogong 2026-09-18 15:54:43 +08:00 committed by GitHub
parent b63a9565e3 9fa620e2e5
commit e2ee878519
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
eval_llm.py
View File

@ -64,7 +64,14 @@ def main():
input_mode = int(input('[0] 自动测试\n[1] 手动输入\n')) input_mode = int(input('[0] 自动测试\n[1] 手动输入\n'))
streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True) streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
prompt_iter = prompts if input_mode == 0 else iter(lambda: input('💬: '), '') def _next_prompt():
try:
return input('💬: ')
except (EOFError, KeyboardInterrupt):
print()
return ''
prompt_iter = prompts if input_mode == 0 else iter(_next_prompt, '')
for prompt in prompt_iter: for prompt in prompt_iter:
setup_seed(random.randint(0, 31415926)) setup_seed(random.randint(0, 31415926))
if input_mode == 0: print(f'💬: {prompt}') if input_mode == 0: print(f'💬: {prompt}')