From 17199b8678f3be962d9021f26ab378e22ca3f2fa Mon Sep 17 00:00:00 2001
From: wangzy <wangziyi@pjlab.org.cn>
Date: Tue, 6 Feb 2024 11:11:57 +0800
Subject: [PATCH] fix arg type

---
 agent/streaming_inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/agent/streaming_inference.py b/agent/streaming_inference.py
index e5532b7..fc36b34 100644
--- 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')