Commit af45872f authored by aria1th's avatar aria1th
Browse files

copy LDM VAE key from XL

parent b29fc6d4

modules/hypertile.py

+1 −0
Original line number Diff line number Diff line
@@ -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)

CRA Git | Maintained and supported by SUSTech CRA and CCSE