'commit'

dsLightRag/Liblib/T1.py

@@ -1,4 +1,4 @@
-from .LibLibUtil import LibLibUtil
+from LibLibUtil import LibLibUtil
 
 
 if __name__ == '__main__':
@@ -15,24 +15,4 @@ if __name__ == '__main__':
         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("图片生成失败")
+        print("获取模型信息失败")