diff --git a/comfy/model_downloader.py b/comfy/model_downloader.py index bf57afc6c..3189f1dce 100644 --- a/comfy/model_downloader.py +++ b/comfy/model_downloader.py @@ -225,10 +225,8 @@ KNOWN_CHECKPOINTS: Final[KnownDownloadables] = KnownDownloadables([ HuggingFile("stabilityai/stable-cascade", "comfyui_checkpoints/stable_cascade_stage_b.safetensors"), HuggingFile("stabilityai/stable-cascade", "comfyui_checkpoints/stable_cascade_stage_c.safetensors"), HuggingFile("stabilityai/stable-cascade", "comfyui_checkpoints/stage_a.safetensors", show_in_ui=False), - HuggingFile("runwayml/stable-diffusion-v1-5", "v1-5-pruned-emaonly.safetensors"), - HuggingFile("runwayml/stable-diffusion-v1-5", "v1-5-pruned-emaonly.ckpt", show_in_ui=False), - HuggingFile("runwayml/stable-diffusion-v1-5", "v1-5-pruned.ckpt", show_in_ui=False), - HuggingFile("runwayml/stable-diffusion-v1-5", "v1-5-pruned.safetensors", show_in_ui=False), + HuggingFile("Comfy-Org/stable-diffusion-v1-5-archive", "v1-5-pruned-emaonly.safetensors"), + HuggingFile("Comfy-Org/stable-diffusion-v1-5-archive", "v1-5-pruned-emaonly-fp16.safetensors"), # from https://github.com/comfyanonymous/ComfyUI_examples/tree/master/2_pass_txt2img HuggingFile("stabilityai/stable-diffusion-2-1", "v2-1_768-ema-pruned.ckpt", show_in_ui=False), HuggingFile("waifu-diffusion/wd-1-5-beta3", "wd-illusion-fp16.safetensors", show_in_ui=False), diff --git a/docs/examples/script_examples/basic_api_example.py b/docs/examples/script_examples/basic_api_example.py index 34bf9ce08..8758322e6 100644 --- a/docs/examples/script_examples/basic_api_example.py +++ b/docs/examples/script_examples/basic_api_example.py @@ -40,7 +40,7 @@ _PROMPT_FROM_WEB_UI = { "4": { "class_type": "CheckpointLoaderSimple", "inputs": { - "ckpt_name": "v1-5-pruned-emaonly.ckpt" + "ckpt_name": "v1-5-pruned-emaonly.safetensors" } }, "5": { diff --git a/docs/examples/script_examples/remote_api_example.py b/docs/examples/script_examples/remote_api_example.py index db1f93589..9cf183bb3 100644 --- a/docs/examples/script_examples/remote_api_example.py +++ b/docs/examples/script_examples/remote_api_example.py @@ -42,7 +42,7 @@ _PROMPT_FROM_WEB_UI = { "4": { "class_type": "CheckpointLoaderSimple", "inputs": { - "ckpt_name": "v1-5-pruned-emaonly.ckpt" + "ckpt_name": "v1-5-pruned-emaonly.safetensors" } }, "5": {