From a592b1cd3fd8027b392231f09b4e1d91b3cb0346 Mon Sep 17 00:00:00 2001 From: dengpeng777 Date: Sat, 30 Mar 2024 10:00:55 +0800 Subject: [PATCH] fix add default embed model --- server/model_workers/zhipu.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/server/model_workers/zhipu.py b/server/model_workers/zhipu.py index b925760d..1eb37e69 100644 --- a/server/model_workers/zhipu.py +++ b/server/model_workers/zhipu.py @@ -39,6 +39,8 @@ def generate_token(apikey: str, exp_seconds: int): class ChatGLMWorker(ApiModelWorker): + DEFAULT_EMBED_MODEL = "embedding-2" + def __init__( self, *,