From af45872fdb8a66ffd6a405d99120e0bacbb4a170 Mon Sep 17 00:00:00 2001
From: aria1th <35677394+aria1th@users.noreply.github.com>
Date: Wed, 15 Nov 2023 15:15:14 +0900
Subject: copy LDM VAE key from XL

---
 modules/hypertile.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'modules')

diff --git a/modules/hypertile.py b/modules/hypertile.py
index ab1c74c0..32d8604c 100644
--- a/modules/hypertile.py
+++ b/modules/hypertile.py
@@ -35,6 +35,7 @@ DEPTH_LAYERS = {
         "output_blocks.11.1.transformer_blocks.0.attn1",
         # SD 1.5 VAE
         "decoder.mid_block.attentions.0",
+        "decoder.mid.attn_1",
     ],
     1: [
         # SD 1.5 U-Net (diffusers)
-- 
cgit v1.2.1