Skip to content

Commit 61b8c88

Browse files
committed
更新图像处理模块,添加对“zimage”模型的特殊处理逻辑,优化提示词解析流程,以提升图像生成的灵活性和一致性。
1 parent 81e1e19 commit 61b8c88

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

handlers/commands/image.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,12 @@ async def image(msg: types.Message):
343343

344344
# 解析选项和优化提示词
345345
prompt, options = PromptProcessor.parse_options(raw_prompt)
346-
prompt, reply_content = await PromptProcessor.optimize_prompt(prompt)
346+
347+
if options.get("model", DEFAULT_MODEL) == "zimage":
348+
prompt = prompt
349+
reply_content = prompt
350+
else:
351+
prompt, reply_content = await PromptProcessor.optimize_prompt(prompt)
347352

348353
# 创建任务
349354
task = Task(

0 commit comments

Comments
 (0)