From 6faaaf7b3e6d9fd1ca10fc1fbb24ef4e60da7c84 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sat, 3 Feb 2024 20:02:14 +0800 Subject: [PATCH] fix maxlen --- src/imitater/__init__.py | 2 +- src/imitater/model/chat_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/imitater/__init__.py b/src/imitater/__init__.py index d3ec452..3ced358 100644 --- 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 index 4a7a495..86a5f3e 100644 --- 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)