github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix arg type

pull/695/head
wangzy 2024-02-06 11:11:57 +08:00
parent 206c3539dc
commit 17199b8678
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
agent/streaming_inference.py
View File

@ -94,7 +94,7 @@ def parse_args():
default=None, default=None,
help='Parameter for Top-K Sampling.') help='Parameter for Top-K Sampling.')
parser.add_argument('--stop_words', parser.add_argument('--stop_words',
type=list, type=str,
default=['<|action_end|>', '<|im_end|>'], default=['<|action_end|>', '<|im_end|>'],
action='append', action='append',
help='Stop words') help='Stop words')