'commit'

dsLightRag/Liblib/T4_WenShengTu_CustomCheckPoint.py

@@ -1,39 +1,36 @@
 from Liblib.LibLibGenerator import LibLibGenerator
 
 if __name__ == '__main__':
-    generate_params = {
-    "templateUuid": "e10adc3949ba59abbe56e057f20f883e",
-    "generateParams": {
-        "checkPointId": "0ea388c7eb854be3ba3c6f65aac6bfd3", # 底模 modelVersionUUID
-        "prompt": "Asian portrait,A young woman wearing a green baseball cap,covering one eye with her hand", # 选填
-        "negativePrompt": "ng_deepnegative_v1_75t,(badhandv4:1.2),EasyNegative,(worst quality:2),", #选填
-        "sampler": 15, # 采样方法
-        "steps": 20, # 采样步数
-        "cfgScale": 7, # 提示词引导系数
-        "width": 768, # 宽
-        "height": 1024, # 高
-        "imgCount": 1, # 图片数量    
-        "randnSource": 0,  # 随机种子生成器 0 cpu，1 Gpu
-        "seed": 2228967414, # 随机种子值，-1表示随机    
-        "restoreFaces": 0,  # 面部修复，0关闭，1开启
-        
-        # 高分辨率修复
-        "hiResFixInfo": {
-            "hiresSteps": 20, # 高分辨率修复的重绘步数
-            "hiresDenoisingStrength": 0.75, # 高分辨率修复的重绘幅度
-            "upscaler": 10, # 放大算法模型枚举
-            "resizedWidth": 1024,  # 放大后的宽度
-            "resizedHeight": 1536  # 放大后的高度
-        }
-    }
-}
-
     liblib = LibLibGenerator()
-
-    # 调用生成接口
-    response = liblib.post_request(
-        "/api/generate/webui/text2img",
-        generate_params
+    
+    # 设置高分辨率修复参数
+    hi_res_fix_info = {
+        "hiresSteps": 20,
+        "hiresDenoisingStrength": 0.75,
+        "upscaler": 10,
+        "resizedWidth": 1024,
+        "resizedHeight": 1536
+    }
+    
+    # 调用自定义Checkpoint方法进行文生图
+    result = liblib.generate_custom_checkpoint_text_to_image(
+        template_uuid="e10adc3949ba59abbe56e057f20f883e",
+        checkpoint_id="0ea388c7eb854be3ba3c6f65aac6bfd3",
+        prompt="Asian portrait,A young woman wearing a green baseball cap,covering one eye with her hand",
+        negative_prompt="ng_deepnegative_v1_75t,(badhandv4:1.2),EasyNegative,(worst quality:2),",
+        sampler=15,
+        steps=20,
+        cfg_scale=7,
+        width=768,
+        height=1024,
+        img_count=1,
+        randn_source=0,
+        seed=2228967414,
+        restore_faces=0,
+        hi_res_fix_info=hi_res_fix_info
     )
-    # {'generateUuid': 'f16463d55e144214ae12d5177f8256fc'}
-    print(response)
+    
+    if result:
+        print(f"🎉 生图任务完成！最终OBS地址: {result}")
+    else:
+        print("❌ 生图任务失败")