'commit'
This commit is contained in:
@@ -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:
|
||||||
|
@@ -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)
|
||||||
|
Binary file not shown.
21
dsLightRag/Liblib/参数模板预设.txt
Normal file
21
dsLightRag/Liblib/参数模板预设.txt
Normal 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 - 仅用于人像换脸
|
||||||
|
- 注意人像参考图中的人物面部特征务必清晰
|
Reference in New Issue
Block a user