Update nodes.py

2024-08-22 23:07:20 +03:00
parent 9b59d91964
commit 4ec6d50315
1 changed files with 2 additions and 1 deletions
@@ -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,