mirror of https://github.com/Mai-with-u/MaiBot.git
Update utils_image.py
parent
13008a14be
commit
9fdfb71801
|
|
@ -273,7 +273,7 @@ class ImageManager:
|
|||
prompt = global_config.personality.visual_style
|
||||
logger.info(f"[VLM调用] 为图片生成新描述 (Hash: {image_hash[:8]}...)")
|
||||
description, _ = await self.vlm.generate_response_for_image(
|
||||
prompt, image_base64, image_format, temperature=0.4
|
||||
prompt, image_base64, image_format, temperature=0.4, max_tokens = model_config.model_task_config.vlm.img_tokens if max_tokens is None else max_tokens
|
||||
)
|
||||
|
||||
if description is None:
|
||||
|
|
@ -570,7 +570,7 @@ class ImageManager:
|
|||
|
||||
# 获取VLM描述
|
||||
description, _ = await self.vlm.generate_response_for_image(
|
||||
prompt, image_base64, image_format, temperature=0.4
|
||||
prompt, image_base64, image_format, temperature=0.4, max_tokens = model_config.model_task_config.vlm.img_tokens if max_tokens is None else max_tokens
|
||||
)
|
||||
|
||||
if description is None:
|
||||
|
|
|
|||
Loading…
Reference in New Issue