1

2023-03-21 21:35:41 +09:00
parent 782c329ae9
commit 8d14509bbe
8 changed files with 395 additions and 164 deletions
@@ -4,18 +4,22 @@ from nodes import *
 import folder_paths
 import random
 from .check_name import *
-
-print(os.getcwd())
+from .Colors import cprint, Colors as Colors
+import os
+py_name=os.path.basename(__file__)
+cprint(py_name, Colors.BLUE)

 class CheckpointLoaderSimpleText:
    @classmethod
    def INPUT_TYPES(s):
+        t_checkpoints=folder_paths.get_filename_list("checkpoints")
+        #cprint(f"checkpoints count : {len(t_checkpoints)}", Colors.BGREEN)
        return {
            "required": { 
                "ckpt_name": (
                    "STRING", {
                        "multiline": False, 
-                        "default": random.choice(folder_paths.get_filename_list("checkpoints"))
+                        "default": random.choice(t_checkpoints)
                    }
                ),
             }