类功能

功能描述

对接文生图大模型服务，提供大模型交互功能。

当前只支持模型：stable-diffusion-v1-5和stable-diffusion-2-1-base。

函数原型

from mx_rag.llm import Text2ImgMultiModel
Text2ImgMultiModel(url, model_name, client_param)

输入参数说明

参数名	数据类型	可选/必选	说明
url	str	必选	大模型访问url。长度取值[1, 128]。
model_name	str	可选	SD模型名称。默认值为“None”。长度取值范围(0,128]。
client_param	ClientParam	可选	https客户端配置参数，默认值为“ClientParam()”，具体描述请参见ClientParam类

返回值说明

Text2ImgMultiModel对象。

调用示例

from mx_rag.llm import Text2ImgMultiModel
from mx_rag.utils import ClientParam
multi_model = Text2ImgMultiModel(model_name="sd", url="txt to image url",
                                 client_param=ClientParam(ca_file="/path/to/ca.crt"))
res = multi_model.text2img(prompt="dog wearing black glasses", output_format="jpg", size="512*512")
print(res)

父主题： Text2ImgMultiModel类