copy LDM VAE key from XL

af45872f · aria1th · b29fc6d4 · af45872f
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

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

未找到文件。
--- 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)