1

2023-03-20 17:26:23 +09:00
parent 479025c947
commit 0ac73701c6
12 changed files with 566 additions and 1 deletions
@@ -0,0 +1,43 @@
+import os
+import comfy.sd
+from nodes import *
+import folder_paths
+import random
+from .check_name import *
+
+print(os.getcwd())
+
+class CheckpointLoaderSimpleText:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": { 
+                "ckpt_name": (
+                    "STRING", {
+                        "multiline": False, 
+                        "default": random.choice(folder_paths.get_filename_list("checkpoints"))
+                    }
+                ),
+             }
+         }
+    RETURN_TYPES = ("MODEL", "CLIP", "VAE")
+    FUNCTION = "load_checkpoint"
+
+    CATEGORY = "loaders"
+    
+    def load_checkpoint(self, ckpt_name, output_vae=True, output_clip=True):
+        ckpt_path =check_name_ckpt(name_split_choice(ckpt_name))
+        if ckpt_path is None:
+            print(f"{ckpt_name} is none")
+            return 
+        try:
+            out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings"))
+            return out
+        except Exception as e:
+            print("CheckpointLoaderSimpleText Exception : "+ e)
+            return 
+            
+        
+#NODE_CLASS_MAPPINGS = {
+#    "CheckpointLoaderSimpleText": CheckpointLoaderSimpleText,
+#}