From 5151d5b0173c990839a6e08e97c080042f4867fd Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 31 May 2024 16:21:10 -0700
Subject: [PATCH] Include HyperSD checkpoints

---
 comfy/model_downloader.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/comfy/model_downloader.py b/comfy/model_downloader.py
index ce4aebf45..d6af7b7a7 100644
--- a/comfy/model_downloader.py
+++ b/comfy/model_downloader.py
@@ -195,6 +195,8 @@ KNOWN_CLIP_VISION_MODELS = [
 KNOWN_LORAS = [
     CivitFile(model_id=211577, model_version_id=238349, filename="openxl_handsfix.safetensors"),
     CivitFile(model_id=324815, model_version_id=364137, filename="blur_control_xl_v1.safetensors"),
+    HuggingFile("ByteDance/Hyper-SD", "Hyper-SDXL-12steps-CFG-lora.safetensors"),
+    HuggingFile("ByteDance/Hyper-SD", "Hyper-SD15-12steps-CFG-lora.safetensors")
 ]
 
 KNOWN_CONTROLNETS = [
@@ -303,7 +305,9 @@ KNOWN_HUGGINGFACE_MODEL_REPOS = {
     "microsoft/Phi-3-mini-4k-instruct",
 }
 
-KNOWN_UNET_MODELS: List[Union[CivitFile | HuggingFile]] = []
+KNOWN_UNET_MODELS: List[Union[CivitFile | HuggingFile]] = [
+    HuggingFile("ByteDance/Hyper-SD", "Hyper-SDXL-1step-Unet-Comfyui.fp16.safetensors")
+]
 
 KNOWN_CLIP_MODELS: List[Union[CivitFile | HuggingFile]] = []