import sys cnet_models = ["canny", "normal", "depth", "openpose", "hed", "scribble", "mlsd", "seg"] unet_models = ["sd15", "openjourney", "Nitro-Diffusion", "anything-v3.0", "vintedois-diffusion-v0-1"] print(f"# ControlNet Generation Comparison") print("") def print_header(): print("|seed|", end="") for ot in unet_models: print(f"{ot}|", end="") print("") print("|", end="") for _ in range(len(unet_models) + 1): print("---|", end="") print("") def get_url(path): # url = "./all_controlnet_models/" url = 'https://huggingface.co/takuma104/controlnet_dev/resolve/main/model_compare_full/all_controlnet_models/' return url + path for cnet in cnet_models: print(f"## ControlNet {cnet}") print("") print_header() for seed in range(16): print(f"|{seed}|", end="") for unet in unet_models: url = get_url(path=f"{cnet}/{unet}/output_{seed:02d}.png") print(f'|', end="") print("")