copy LDM VAE key from XL

2023-11-15 15:15:14 +09:00 · 2023-11-15 15:15:14 +09:00 · af45872fdb
parent b29fc6d4de
commit af45872fdb
1 changed files with 1 additions and 0 deletions
--- 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)