diff --git a/py/zhipu_glm4.py b/py/zhipu_glm4.py index 9499be4..7caf967 100644 --- a/py/zhipu_glm4.py +++ b/py/zhipu_glm4.py @@ -4,7 +4,6 @@ import torch import base64 import requests from io import BytesIO -from zhipuai import ZhipuAI import folder_paths from PIL import Image from .imagefunc import log, tensor2pil, get_api_key @@ -34,6 +33,7 @@ class LS_ZhipuImage: CATEGORY = '😺dzNodes/LayerUtility' def zhipu_glm4v(self, image, model, user_prompt,): + from zhipuai import ZhipuAI client = ZhipuAI(api_key=get_api_key('zhipu_api_key')) # APIKey @@ -90,7 +90,7 @@ class LS_ZhipuText: CATEGORY = '😺dzNodes/LayerUtility' def zhipu_glm4(self, model, user_prompt, history_length, history=None): - + from zhipuai import ZhipuAI client = ZhipuAI(api_key=get_api_key('zhipu_api_key')) # APIKey if history is not None: diff --git a/pyproject.toml b/pyproject.toml index b78f78a..ff0fcbe 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,9 +1,9 @@ [project] name = "comfyui_layerstyle_advance" description = "The nodes detached from ComfyUI Layer Style are mainly those with complex requirements for dependency packages." -version = "2.0.9" +version = "2.0.10" license = "MIT" -dependencies = ["numpy", "matplotlib", "scikit_image", "scikit_learn", "opencv-contrib-python", "pymatting", "timm", "blend_modes", "transformers", "diffusers", "loguru", "colour-science", "huggingface_hub", "segment_anything", "addict", "omegaconf", "yapf", "wget", "iopath", "mediapipe", "typer_config", "fastapi", "rich", "google-generativeai", "ultralytics", "transparent-background", "accelerate", "onnxruntime", "bitsandbytes", "peft", "protobuf", "hydra-core", "blind-watermark", "qrcode", "pyzbar", "psd-tools", "wandb"] +dependencies = ["numpy", "matplotlib", "scikit_image", "scikit_learn", "opencv-contrib-python", "pymatting", "timm", "blend_modes", "transformers", "diffusers", "loguru", "colour-science", "huggingface_hub", "segment_anything", "addict", "omegaconf", "yapf", "wget", "iopath", "mediapipe", "typer_config", "fastapi", "rich", "google-generativeai", "ultralytics", "transparent-background", "accelerate", "onnxruntime", "bitsandbytes", "peft", "protobuf", "hydra-core", "blind-watermark", "qrcode", "pyzbar", "psd-tools", "wandb","zhipuai"] [project.urls] Repository = "https://github.com/chflame163/ComfyUI_LayerStyle_Advance" diff --git a/requirements.txt b/requirements.txt index 5fc0f32..5b9633a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -33,4 +33,5 @@ hydra-core blind-watermark qrcode pyzbar -psd-tools \ No newline at end of file +psd-tools +zhipuai \ No newline at end of file