39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from .LibLibUtil import LibLibUtil
|
||
|
||
|
||
if __name__ == '__main__':
|
||
liblib = LibLibUtil()
|
||
|
||
# 测试获取模型版本信息
|
||
print("===== 测试获取模型版本信息 =====")
|
||
version_uuid = "390e7df48aed45d4bc3f0bcbc89fa44c"
|
||
model_info = liblib.get_model_version_info(version_uuid)
|
||
|
||
if model_info:
|
||
print(f"模型名称：{model_info['modelName']}")
|
||
print(f"版本名称：{model_info['versionName']}")
|
||
print(f"是否允许商业用途：{model_info['commercialUse']}")
|
||
print(f"模型下载地址：{model_info['modelUrl']}")
|
||
else:
|
||
print("获取模型信息失败")
|
||
|
||
# 测试生成图片
|
||
print("\n===== 测试生成图片 =====")
|
||
template_uuid = "7d888009f81d4252a7c458c874cd017f"
|
||
generate_params = {
|
||
"checkPointId": "0ea388c7eb854be3ba3c6f65aac6bfd3",
|
||
"prompt": "Asian portrait,A young woman wearing a green baseball cap",
|
||
"width": 768,
|
||
"height": 1152,
|
||
"sampler": 20,
|
||
"steps": 35,
|
||
"cfgScale": 2.0,
|
||
"imgCount": 1
|
||
}
|
||
|
||
image_data = liblib.generate_text_to_image(template_uuid, generate_params)
|
||
if image_data:
|
||
print(f"图片生成成功，任务UUID: {image_data.get('generateUuid')}")
|
||
else:
|
||
print("图片生成失败")
|