Skip to content

Commit

Permalink
fix bug of florence2 model loader
Browse files Browse the repository at this point in the history
  • Loading branch information
chflame163 committed Aug 23, 2024
1 parent 31253df commit 65820e0
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 1 addition & 2 deletions py/imagefunc.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -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"]

Expand Down

0 comments on commit 65820e0

Please sign in to comment.