dsProject/dsLightRag/Liblib/T3.py

import json

from Liblib.LibLibUtil import SAMPLING_METHODS
from PuLIDGenerator import PuLIDGenerator

# 加载配置文件
with open('config.json', 'r', encoding='utf-8') as f:
    config = json.load(f)

if __name__ == "__main__":
    # 获取第一个模型的配置
    model_config = config["models"][1]
    # 完成的例子
    # https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/484eba61d6e74f8ea0cb3d8ff2d14341.jpg

    # 通过枚举转换
    print(f'使用采样方法: {model_config["sampler"]}')
    model_config["sampler"] = SAMPLING_METHODS[model_config["sampler"]]
    # 创建生成器实例
    generator = PuLIDGenerator()

    # 使用配置参数
    generator.set_default_params(
        template_uuid=model_config["template_uuid"],
        steps=model_config["steps"],
        width=model_config["width"],
        height=model_config["height"]
    )

    # 生成图像
    obs_url = generator.generate_image(
        prompt=model_config["prompt"],
        reference_image_url=model_config["reference_image_url"],
        control_weight=model_config["control_weight"]
    )

    if obs_url:
        print(f"最终OBS地址: {obs_url}")