46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
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"][0]
|
|
# 完成的例子
|
|
# 炫酷机甲美女_majicflus
|
|
# https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/bdde20e3143049c3916aa0b8d523b469.jpg
|
|
|
|
# 卡通小人
|
|
# https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/484eba61d6e74f8ea0cb3d8ff2d14341.jpg
|
|
|
|
# F.1-武侠世界
|
|
# https://dsideal.obs.cn-north-1.myhuaweicloud.com/HuangHai/LibLib/215df7724cba4bf9a64e7863ecb4ebc0.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}")
|