From fa3370df5b18b17c5423f088439dcb95b0fdb886 Mon Sep 17 00:00:00 2001 From: zhayujie Date: Tue, 7 May 2024 11:37:27 +0800 Subject: [PATCH] fix: image model check --- common/linkai_client.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/common/linkai_client.py b/common/linkai_client.py index 642c66a..3329f74 100644 --- a/common/linkai_client.py +++ b/common/linkai_client.py @@ -60,6 +60,9 @@ class ChatClient(LinkAIClient): if pconf("linkai")["midjourney"]: pconf("linkai")["midjourney"]["enabled"] = True pconf("linkai")["midjourney"]["use_image_create_prefix"] = True + elif config.get("text_to_image") and config.get("text_to_image") in ["dall-e-2", "dall-e-3"]: + if pconf("linkai")["midjourney"]: + pconf("linkai")["midjourney"]["use_image_create_prefix"] = False def start(channel):