fix arg type

2024-02-06 11:11:57 +08:00 · 2024-02-06 11:11:57 +08:00 · 17199b8678
parent 206c3539dc
commit 17199b8678
1 changed files with 1 additions and 1 deletions
--- a/agent/streaming_inference.py
+++ b/agent/streaming_inference.py
@ -94,7 +94,7 @@ def parse_args():
                        default=None,
                        help='Parameter for Top-K Sampling.')
    parser.add_argument('--stop_words',
-                        type=list,
+                        type=str,
                        default=['<|action_end|>', '<|im_end|>'],
                        action='append',
                        help='Stop words')