From cacb022c4a5b9614f96086a866c8a4c4e9e85760 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 18 Mar 2024 00:26:23 -0400
Subject: [PATCH] Make saved SD1 checkpoints match more closely the official
 one.

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

diff --git a/comfy/supported_models.py b/comfy/supported_models.py
index 3758210..e12935a 100644
--- a/comfy/supported_models.py
+++ b/comfy/supported_models.py
@@ -45,6 +45,11 @@ class SD15(supported_models_base.BASE):
         return state_dict
 
     def process_clip_state_dict_for_saving(self, state_dict):
+        pop_keys = ["clip_l.transformer.text_projection.weight", "clip_l.logit_scale"]
+        for p in pop_keys:
+            if p in state_dict:
+                state_dict.pop(p)
+
         replace_prefix = {"clip_l.": "cond_stage_model."}
         return utils.state_dict_prefix_replace(state_dict, replace_prefix)