fix maxlen

the-seeds · Feb 3, 2024 · 6faaaf7 · 6faaaf7
1 parent f4ec551
commit 6faaaf7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/imitater/__init__.py b/src/imitater/__init__.py
@@ -1 +1 @@
-__version__ = "0.2.0"
+__version__ = "0.2.1"
diff --git a/src/imitater/model/chat_model.py b/src/imitater/model/chat_model.py
@@ -32,7 +32,7 @@ def add_cli_args(parser: "ArgumentParser") -> None:
         parser.add_argument("--path", type=str)
         parser.add_argument("--device", type=int, nargs="+")
         parser.add_argument("--port", type=int)
-        parser.add_argument("--maxlen", type=int, default=4096)
+        parser.add_argument("--maxlen", type=int, default=2048)
         parser.add_argument("--agent_type", type=str, choices=list_agents(), default="react")
         parser.add_argument("--template", type=str, default=None)
         parser.add_argument("--gen_config", type=str, default=None)