diff --git a/dsLightRag/Liblib/LibLibUtil.py b/dsLightRag/Liblib/LibLibUtil.py index 9f8fe6b8..3ebb1340 100644 --- a/dsLightRag/Liblib/LibLibUtil.py +++ b/dsLightRag/Liblib/LibLibUtil.py @@ -114,7 +114,7 @@ class LibLibUtil: print(f"生图任务超时({max_wait_time}秒)") return None - def download_and_upload_to_obs(image_url, generate_uuid): + def download_and_upload_to_obs(slef,image_url, generate_uuid): """下载图片并上传到OBS""" try: # 1. 清理URL(去除可能的引号和空格) diff --git a/dsLightRag/Liblib/T1.py b/dsLightRag/Liblib/T1.py index f8783fdd..c7036707 100644 --- a/dsLightRag/Liblib/T1.py +++ b/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("获取模型信息失败") \ No newline at end of file diff --git a/dsLightRag/Liblib/__pycache__/LibLibUtil.cpython-310.pyc b/dsLightRag/Liblib/__pycache__/LibLibUtil.cpython-310.pyc new file mode 100644 index 00000000..9af891df Binary files /dev/null and b/dsLightRag/Liblib/__pycache__/LibLibUtil.cpython-310.pyc differ