Fix the input validation of the Prompt Reader node #37

receyuki · Jan 2, 2024 · 202b1fd · 202b1fd
1 parent a7fd3f1
commit 202b1fd
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/nodes.py b/nodes.py
@@ -230,17 +230,14 @@ def search_model(model: str):
 
     @classmethod
     def IS_CHANGED(s, image, parameter_index):
-        image_path = folder_paths.get_annotated_filepath(image)
+        if image in SDPromptReader.files:
+            image_path = folder_paths.get_annotated_filepath(image)
+        else:
+            image_path = image
         with open(Path(image_path), "rb") as f:
             image_data = ImageDataReader(f)
         return image_data.props
 
-    @classmethod
-    def VALIDATE_INPUTS(s, image, parameter_index):
-        if not folder_paths.exists_annotated_filepath(image):
-            return "Invalid image file: {}".format(image)
-        return True
-
 
 class SDPromptSaver:
     def __init__(self):