From 65820e056abfbf8396ca7b7dfd8ecc0029b7c576 Mon Sep 17 00:00:00 2001 From: chflame163 Date: Fri, 23 Aug 2024 10:51:32 +0800 Subject: [PATCH] fix bug of florence2 model loader --- py/imagefunc.py | 3 +-- pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/py/imagefunc.py b/py/imagefunc.py index 91ffe91..d3affc6 100644 --- a/py/imagefunc.py +++ b/py/imagefunc.py @@ -30,7 +30,7 @@ from skimage import img_as_float, img_as_ubyte import torchvision.transforms.functional as TF import torch.nn.functional as F -from transformers import StoppingCriteria, StoppingCriteriaList +from transformers import AutoModel, AutoProcessor, StoppingCriteria, StoppingCriteriaList, AutoModelForCausalLM from colorsys import rgb_to_hsv import folder_paths from .filmgrainer import processing as processing_utils @@ -1997,7 +1997,6 @@ class UformGen2QwenChat: def __init__(self): from huggingface_hub import snapshot_download - from transformers import AutoModel, AutoProcessor # self.model_path = snapshot_download("unum-cloud/uform-gen2-qwen-500m", # local_dir=files_for_uform_gen2_qwen, # force_download=False, # Set to True if you always want to download, regardless of local copy diff --git a/pyproject.toml b/pyproject.toml index c70143a..d5b7f58 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui_layerstyle" description = "A set of nodes for ComfyUI it generate image like Adobe Photoshop's Layer Style. the Drop Shadow is first completed node, and follow-up work is in progress." -version = "1.0.25" +version = "1.0.26" license = "MIT" dependencies = ["numpy", "pillow", "torch", "matplotlib", "Scipy", "scikit_image", "opencv-contrib-python", "pymatting", "segment_anything", "timm", "addict", "yapf", "colour-science", "wget", "mediapipe", "loguru", "typer_config", "fastapi", "rich", "google-generativeai", "diffusers", "omegaconf", "tqdm", "transformers", "kornia", "image-reward", "ultralytics", "blend_modes", "blind-watermark", "qrcode", "pyzbar", "transparent-background", "huggingface_hub", "psd-tools"]