Update baichuan.py to support max_tokens

cddc · May 11, 2024 · 8a10765 · 8a10765
1 parent 426f9c4
commit 8a10765
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/generate/chat_completion/models/baichuan.py b/generate/chat_completion/models/baichuan.py
@@ -38,6 +38,7 @@ class BaichuanMessage(TypedDict):
 
 class BaichuanChatParameters(ModelParameters):
     temperature: Optional[Temperature] = None
+    max_tokens: Optional[PositiveInt] = None
     top_k: Optional[Annotated[int, Field(ge=0)]] = None
     top_p: Optional[Probability] = None
     search: Optional[bool] = Field(default=None, alias='with_search_enhance')