This commit is contained in:
2025-09-04 10:43:10 +08:00
parent 4d7bf616e7
commit 83f67f7d54
4 changed files with 54 additions and 44 deletions

View File

@@ -85,6 +85,7 @@ class LibLibUtil:
"templateUuid": template_uuid, "templateUuid": template_uuid,
"generateParams": generate_params "generateParams": generate_params
} }
print(payload)
return self.post_request(uri, payload) return self.post_request(uri, payload)
def get_generation_status(self, generate_uuid): def get_generation_status(self, generate_uuid):
@@ -207,7 +208,6 @@ class LibLibUtil:
""" """
try: try:
# 调用生成接口 # 调用生成接口
response = self.generate_text_to_image(template_uuid, generate_params) response = self.generate_text_to_image(template_uuid, generate_params)
if response and "generateUuid" in response: if response and "generateUuid" in response:

View File

@@ -1,45 +1,34 @@
import json from Liblib.LibLibUtil import LibLibUtil
from LibLibUtil import LibLibUtil, SAMPLING_METHODS
def test_generate_image_with_obs():
"""测试generate_image_with_obs方法"""
liblib = LibLibUtil()
print("===== 测试 generate_image_with_obs 方法 =====")
# 加载配置文件
with open('config.json', 'r', encoding='utf-8') as f:
config = json.load(f)
# 使用第一个模型的配置
first_model = config['models'][0]
# 设置生图参数
template_uuid = first_model["template_uuid"]
generate_params = {
"prompt": first_model["prompt"],
"negative_prompt": first_model["negative_prompt"],
"steps": first_model["steps"],
"width": first_model["width"],
"height": first_model["height"],
"cfg_scale": first_model["cfgScale"],
"sampler_name": first_model["sampler"],
"sampler_index": SAMPLING_METHODS[first_model["sampler"]],
"seed": first_model["seed"]
}
print(f"使用模型: {first_model['name']}")
print(f"模板UUID: {template_uuid}")
# 使用新的generate_image_with_obs方法
file_url = liblib.generate_image_with_obs(template_uuid, generate_params)
if file_url:
print(f"✅ 生图任务完成OBS地址: {file_url}")
return file_url
else:
print("❌ 生图任务失败")
return None
if __name__ == '__main__': if __name__ == '__main__':
# 测试新的方法 """
test_generate_image_with_obs()
"""
generate_params = {
"templateUuid": "6f7c4652458d4802969f8d089cf5b91f", # 参数模板ID F.1文生图
"generateParams": {
"prompt": "filmfotos, Asian portrait,A young woman wearing a green baseball cap,covering one eye with her hand",
"steps": 20, # 采样步数
"width": 768, # 宽
"height": 1024, # 高
"imgCount": 1, # 图片数量
"seed": -1, # 随机种子值,-1 表示随机
"restoreFaces": 0, # 面部修复0 关闭1 开启
# Lora添加最多5个不添加lora时请删除此结构体
"additionalNetwork": [
{
"modelId": "169505112cee468b95d5e4a5db0e5669", # LoRA的模型版本uuid
"weight": 1.0 # LoRA权重
}
]
}
}
liblib = LibLibUtil()
# 调用生成接口
response = liblib.post_request(
"/api/generate/webui/text2img",
generate_params
)
# {'generateUuid': 'f16463d55e144214ae12d5177f8256fc'}
print(response)

View File

@@ -0,0 +1,21 @@
适用方向 模板名称 模板UUID 备注
F.1文生图 F.1文生图 - 自定义完整参数 6f7c4652458d4802969f8d089cf5b91f - Checkpoint默认为官方模型
- 可用模型范围基础算法F.1
- 支持additional network
F.1图生图 F.1图生图 - 自定义完整参数 63b72710c9574457ba303d9d9b8df8bd - Checkpoint默认为官方模型
- 可用模型范围基础算法F.1
- 支持additional network
1.5和XL文生图 1.5和XL文生图 - 自定义完整参数 e10adc3949ba59abbe56e057f20f883e - 可用模型范围基础算法1.5基础算法XL
- 支持additional network高分辨率修复和controlnet
- 可通过自由拼接参数实现各类的文生图诉求
1.5和XL图生图 1.5和XL图生图 - 自定义完整参数 9c7d531dc75f476aa833b3d452b8f7ad - 可用模型范围基础算法1.5基础算法XL
- 支持additional network和controlnet
- 可通过自由拼接参数实现各类的图生图和蒙版重绘诉求
局部重绘 Controlnet局部重绘 b689de89e8c9407a874acd415b3aa126 - 提取自文生图完整参数
- 支持additional network和controlnet
- 不支持高分辨率修复hiresfix
局部重绘 图生图局部重绘 74509e1b072a4c45a7f1843a963c8462 - 提取自图生图完整参数
- 支持additionalNetwork
- 不支持Controlnet
人物换脸 InstantID人像换脸 7d888009f81d4252a7c458c874cd017f - 仅用于人像换脸
- 注意人像参考图中的人物面部特征务必清晰