| | import os |
| | import comfy.sd |
| | from nodes import * |
| | import random |
| | import folder_paths |
| |
|
| | cnt=0 |
| | ckpt_name="" |
| | ckpt_path="" |
| |
|
| | class CheckpointLoaderRandom: |
| | models_dir = os.path.join(os.getcwd(),"ComfyUI", "models") |
| | ckpt_dir = os.path.join(models_dir, "checkpoints") |
| | cnt=0 |
| | ckpt_name="" |
| | ckpt_path="" |
| | |
| | def __init__(self): |
| | print(f"CheckpointLoaderRandom __init__") |
| | pass |
| | |
| | @classmethod |
| | def INPUT_TYPES(s): |
| | return { |
| | "required": { |
| | "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), |
| | "max": ("INT", {"default": 10, "min": 0, "max": 0xffffffffffffffff}), |
| | |
| | } |
| | } |
| | RETURN_TYPES = ("MODEL", "CLIP", "VAE") |
| | FUNCTION = "load_checkpoint" |
| |
|
| | CATEGORY = "loaders" |
| |
|
| | def load_checkpoint(self, seed, max, output_vae=True, output_clip=True): |
| | global cnt, ckpt_name, ckpt_path |
| | print(f"cnt : { cnt}") |
| | if ckpt_name=="" or cnt>=max : |
| | cnt=0 |
| | ckpt_names= folder_paths.get_filename_list("checkpoints") |
| | |
| | ckpt_name=random.choice(ckpt_names) |
| | print(f"ckpt_name : { ckpt_name}") |
| | ckpt_path = folder_paths.get_full_path("checkpoints", ckpt_name) |
| | out = comfy.sd.load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, embedding_directory=folder_paths.get_folder_paths("embeddings")) |
| | cnt+=1 |
| | return out |
| | |
| |
|
