From ce56af0c41bd494288abc8149c55f2e7c114a712 Mon Sep 17 00:00:00 2001 From: chflame163 Date: Fri, 6 Sep 2024 17:49:56 +0800 Subject: [PATCH] Florence2Image2Prompt support base-PromptGen-v1.5 model --- py/florence2_ultra.py | 3 ++- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/py/florence2_ultra.py b/py/florence2_ultra.py index 2dfa4ca..fe0fc47 100644 --- a/py/florence2_ultra.py +++ b/py/florence2_ultra.py @@ -21,7 +21,8 @@ fl2_model_repos = { "SD3-Captioner": "gokaygokay/Florence-2-SD3-Captioner", "base-PromptGen": "MiaoshouAI/Florence-2-base-PromptGen", "CogFlorence-2-Large-Freeze": "thwri/CogFlorence-2-Large-Freeze", - "CogFlorence-2.1-Large": "thwri/CogFlorence-2.1-Large" + "CogFlorence-2.1-Large": "thwri/CogFlorence-2.1-Large", + "base-PromptGen-v1.5":"MiaoshouAI/Florence-2-base-PromptGen-v1.5" } diff --git a/pyproject.toml b/pyproject.toml index b4598a7..ca6e785 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.50" +version = "1.0.51" 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", "accelerate", "bitsandbytes", "torchscale", "wandb", "hydra-core", "psd-tools", "inference-cli[yolo-world]", "inference-gpu[yolo-world]", "onnxruntime"]