From 40902acc28d2ed0dbc5fa182735aff28041b4f40 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Tue, 15 Oct 2024 10:13:13 -0700
Subject: [PATCH] Use the HuggingFace file for dreamshaper

---
 comfy/model_downloader.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/model_downloader.py b/comfy/model_downloader.py
index b566b1cf2..ac9e769b3 100644
--- a/comfy/model_downloader.py
+++ b/comfy/model_downloader.py
@@ -246,7 +246,7 @@ KNOWN_CHECKPOINTS: Final[KnownDownloadables] = KnownDownloadables([
     CivitFile(139562, 344487, filename="realvisxlV40_v40Bakedvae.safetensors"),
     HuggingFile("SG161222/Realistic_Vision_V6.0_B1_noVAE", "Realistic_Vision_V6.0_NV_B1_fp16.safetensors"),
     HuggingFile("SG161222/Realistic_Vision_V5.1_noVAE", "Realistic_Vision_V5.1_fp16-no-ema.safetensors"),
-    CivitFile(4384, 128713, filename="dreamshaper_8.safetensors"),
+    HuggingFile("Lykon/DreamShaper", "DreamShaper_8_pruned.safetensors", save_with_filename="dreamshaper_8.safetensors", alternate_filenames=("DreamShaper_8_pruned.safetensors")),
     CivitFile(7371, 425083, filename="revAnimated_v2Rebirth.safetensors"),
     CivitFile(4468, 57618, filename="counterfeitV30_v30.safetensors"),
     CivitFile(241415, 272376, filename="picxReal_10.safetensors"),