From 0905db846d60329c5400ab7635d46e97f0766f02 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Sat, 20 Jul 2024 12:02:59 +0300 Subject: [PATCH] Update mask_nodes.py --- nodes/mask_nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes/mask_nodes.py b/nodes/mask_nodes.py index 721f524..bb2065e 100644 --- a/nodes/mask_nodes.py +++ b/nodes/mask_nodes.py @@ -166,11 +166,11 @@ to ComfyUI/models/clip_seg def segment_image(self, model): from transformers import CLIPSegProcessor, CLIPSegForImageSegmentation - checkpoint_path = os.path.join(folder_paths.models_dir,'clip_seg', model) + checkpoint_path = os.path.join(folder_paths.models_dir,'clip_seg', os.path.basename(model)) if not hasattr(self, "model"): if not os.path.exists(checkpoint_path): from huggingface_hub import snapshot_download - snapshot_download(repo_id=model, local_dir=checkpoint_path.split("/")[-1], local_dir_use_symlinks=False) + snapshot_download(repo_id=model, local_dir=checkpoint_path, local_dir_use_symlinks=False) self.model = CLIPSegForImageSegmentation.from_pretrained(checkpoint_path) processor = CLIPSegProcessor.from_pretrained(checkpoint_path)