fix import issue on Transformers 5.x

This commit is contained in:
chflame163
2026-01-30 16:31:44 +08:00
parent f16192bfa7
commit f0fe35bd6d
2 changed files with 7 additions and 2 deletions
+6 -1
View File
@@ -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)
+1 -1
View File
@@ -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"]