From 5379051d1629c1c0849e8292a7bb8717d0bfe749 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 26 Jul 2023 18:26:39 -0400
Subject: [PATCH] Fix diffusers VAE loading.

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

diff --git a/comfy/diffusers_convert.py b/comfy/diffusers_convert.py
index 9688cbd52..a9eb9302f 100644
--- a/comfy/diffusers_convert.py
+++ b/comfy/diffusers_convert.py
@@ -148,6 +148,10 @@ vae_conversion_map_attn = [
     ("q.", "query."),
     ("k.", "key."),
     ("v.", "value."),
+    ("q.", "to_q."),
+    ("k.", "to_k."),
+    ("v.", "to_v."),
+    ("proj_out.", "to_out.0."),
     ("proj_out.", "proj_attn."),
 ]