fix import issue on Transformers 5.x
This commit is contained in:
+6
-1
@@ -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
@@ -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"]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user