| 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 |
| |
|
|
