copy LDM VAE key from XL

af45872f · aria1th · b29fc6d4 · af45872f
Commit af45872f authored Nov 15, 2023 by aria1th
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

modules/hypertile.py modules/hypertile.py +1 -0

No files found.
--- 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)