From f0fe35bd6de6ca3f5b0645b6cf75536897db84ff Mon Sep 17 00:00:00 2001 From: chflame163 Date: Fri, 30 Jan 2026 16:31:44 +0800 Subject: [PATCH] fix import issue on Transformers 5.x --- py/smol.py | 7 ++++++- pyproject.toml | 2 +- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/py/smol.py b/py/smol.py index d736125..05b5317 100644 --- a/py/smol.py +++ b/py/smol.py @@ -4,7 +4,6 @@ import torch import re import folder_paths from PIL import Image -from transformers import AutoProcessor, AutoModelForVision2Seq, AutoTokenizer, AutoModelForCausalLM from .imagefunc import log, check_and_download_model, tensor2pil @@ -60,6 +59,7 @@ class LS_Load_SmolLM2_Model: CATEGORY = '😺dzNodes/LayerUtility' def load_smollm2_model(self, model, dtype, device): + from transformers import AutoTokenizer, AutoModelForCausalLM repo_id = smollm2_repo[model] model_path = os.path.join("smol", model) model_path = check_and_download_model(model_path, repo_id) @@ -96,6 +96,11 @@ class LS_Load_SmolVLM_Model: CATEGORY = '😺dzNodes/LayerUtility' def load_smolvlm_model(self, model, dtype, device): + from transformers import AutoProcessor + try: + from transformers import AutoModelForVision2Seq + except ImportError: + from transformers import AutoModelForImageTextToText as AutoModelForVision2Seq repo_id = smolvlm_repo[model] model_path = os.path.join("smol", model) diff --git a/pyproject.toml b/pyproject.toml index aba3865..d19dc4d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [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.36" +version = "2.0.37" license = { text = "MIT License" } 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", "openai","google-genai", "fastapi","typer-config"]