From c97be4db91d4a249c19afdf88fa1cf3268544e45 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Thu, 30 Nov 2023 19:27:03 -0500
Subject: [PATCH] Support SD2.1 turbo checkpoint.

---
 comfy/supported_models.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/supported_models.py b/comfy/supported_models.py
index 7e2ac677..455323b9 100644
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -71,6 +71,10 @@ class SD20(supported_models_base.BASE):
         return model_base.ModelType.EPS
 
     def process_clip_state_dict(self, state_dict):
+        replace_prefix = {}
+        replace_prefix["conditioner.embedders.0.model."] = "cond_stage_model.model." #SD2 in sgm format
+        state_dict = utils.state_dict_prefix_replace(state_dict, replace_prefix)
+
         state_dict = utils.transformers_convert(state_dict, "cond_stage_model.model.", "cond_stage_model.clip_h.transformer.text_model.", 24)
         return state_dict