diff --git a/nodes/vlm/ram_LP.py b/nodes/vlm/ram_LP.py index a487f88..aa5d906 100644 --- a/nodes/vlm/ram_LP.py +++ b/nodes/vlm/ram_LP.py @@ -4,20 +4,6 @@ import gc import numpy as np import warnings warnings.filterwarnings("ignore", category=FutureWarning, module="timm\\.models") -import sys -import transformers -import types - -wrapper = types.ModuleType("transformers.modeling_utils") - -for k, v in transformers.pytorch_utils.__dict__.items(): - setattr(wrapper, k, v) - -for k, v in transformers.modeling_utils.__dict__.items(): - if not hasattr(wrapper, k): - setattr(wrapper, k, v) - -sys.modules["transformers.modeling_utils"] = wrapper from ram.models import ram from ram.models import ram_plus diff --git a/pyproject.toml b/pyproject.toml index 181972e..fe40db0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-levelpixel-advanced" description = "Advanced nodes of the Level Pixel company (aka levelpixel, LP). Includes convenient advanced nodes for working with LLM и VLM models (LLaVa, Multimodal Generator) with GGUF format. Qwen2.5-VL and Qwen2.5 supported. Also included is a node for the RAM model. Nodes have the ability to automatically unload models from VRAM. This package also includes Autotagger (WD), Image Remove Background (based on RMBG-2.0, on BiRefNet, on BiRefNet-HR-Matter, on RemBG). Licensed under GNU GPLv3. Repository: https://github.com/LevelPixel/ComfyUI-LevelPixel-Advanced" -version = "1.3.5" +version = "1.3.6" license = { file = "LICENSE.txt" } dependencies = [ "torch>=2.0.1", diff --git a/requirements.txt b/requirements.txt index eb8d192..4fc6114 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,4 +14,4 @@ matplotlib diskcache>=5.6.1 jinja2>=2.11.3 typing-extensions>=4.5.0 -ram @ git+https://github.com/xinyu1205/recognize-anything.git +ram @ git+https://github.com/Level-Pixel/recognize-anything.git