From 4ec6d50315092c35ebb336b3e0567bab66fb2ff8 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 22 Aug 2024 23:07:20 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index c1e18b0..8366877 100644 --- a/nodes.py +++ b/nodes.py @@ -14,7 +14,7 @@ import hashlib import warnings import comfy.model_management as mm import folder_paths -from comfy_execution.graph_utils import GraphBuilder + from comfy.utils import ProgressBar @@ -255,6 +255,7 @@ class OneVisionCaptionFolder: img_tensor = img_tensor.unsqueeze(0).permute(0, 2, 3, 1) except IOError: print(f"Cannot open image: {img_path}") + continue result, = vision_node.run( llava_model=llava_model,