imaginAIry/imaginairy/weight_management/weight_info/stable-diffusion-1-5_unet_diffusers.prefixes.json
Bryce f97f6a3b4b feature: use refiners library for generation
BREAKING CHANGE

  - stable diffusion 1.5 + inpainting working
  - self-attention guidance working. improves image generation quality
  - tile-mode working
  - inpainting self-attention guidance working

disable/broken features:
  - sd 1.4, 2.0, 2.1
  - most of the samplers
  - pix2pix edit
  - most of the controlnets
  - memory management
  - python 3.8 support

wip
2023-11-22 13:22:00 -08:00

393 lines
19 KiB
JSON

[
"conv_in",
"conv_norm_out",
"conv_out",
"down_blocks.0.attentions.0.norm",
"down_blocks.0.attentions.0.proj_in",
"down_blocks.0.attentions.0.proj_out",
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k",
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0",
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q",
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v",
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k",
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0",
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q",
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v",
"down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj",
"down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2",
"down_blocks.0.attentions.0.transformer_blocks.0.norm1",
"down_blocks.0.attentions.0.transformer_blocks.0.norm2",
"down_blocks.0.attentions.0.transformer_blocks.0.norm3",
"down_blocks.0.attentions.1.norm",
"down_blocks.0.attentions.1.proj_in",
"down_blocks.0.attentions.1.proj_out",
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k",
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0",
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q",
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v",
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k",
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0",
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q",
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v",
"down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj",
"down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2",
"down_blocks.0.attentions.1.transformer_blocks.0.norm1",
"down_blocks.0.attentions.1.transformer_blocks.0.norm2",
"down_blocks.0.attentions.1.transformer_blocks.0.norm3",
"down_blocks.0.downsamplers.0.conv",
"down_blocks.0.resnets.0.conv1",
"down_blocks.0.resnets.0.conv2",
"down_blocks.0.resnets.0.norm1",
"down_blocks.0.resnets.0.norm2",
"down_blocks.0.resnets.0.time_emb_proj",
"down_blocks.0.resnets.1.conv1",
"down_blocks.0.resnets.1.conv2",
"down_blocks.0.resnets.1.norm1",
"down_blocks.0.resnets.1.norm2",
"down_blocks.0.resnets.1.time_emb_proj",
"down_blocks.1.attentions.0.norm",
"down_blocks.1.attentions.0.proj_in",
"down_blocks.1.attentions.0.proj_out",
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
"down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
"down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
"down_blocks.1.attentions.0.transformer_blocks.0.norm1",
"down_blocks.1.attentions.0.transformer_blocks.0.norm2",
"down_blocks.1.attentions.0.transformer_blocks.0.norm3",
"down_blocks.1.attentions.1.norm",
"down_blocks.1.attentions.1.proj_in",
"down_blocks.1.attentions.1.proj_out",
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
"down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
"down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
"down_blocks.1.attentions.1.transformer_blocks.0.norm1",
"down_blocks.1.attentions.1.transformer_blocks.0.norm2",
"down_blocks.1.attentions.1.transformer_blocks.0.norm3",
"down_blocks.1.downsamplers.0.conv",
"down_blocks.1.resnets.0.conv1",
"down_blocks.1.resnets.0.conv2",
"down_blocks.1.resnets.0.conv_shortcut",
"down_blocks.1.resnets.0.norm1",
"down_blocks.1.resnets.0.norm2",
"down_blocks.1.resnets.0.time_emb_proj",
"down_blocks.1.resnets.1.conv1",
"down_blocks.1.resnets.1.conv2",
"down_blocks.1.resnets.1.norm1",
"down_blocks.1.resnets.1.norm2",
"down_blocks.1.resnets.1.time_emb_proj",
"down_blocks.2.attentions.0.norm",
"down_blocks.2.attentions.0.proj_in",
"down_blocks.2.attentions.0.proj_out",
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
"down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
"down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
"down_blocks.2.attentions.0.transformer_blocks.0.norm1",
"down_blocks.2.attentions.0.transformer_blocks.0.norm2",
"down_blocks.2.attentions.0.transformer_blocks.0.norm3",
"down_blocks.2.attentions.1.norm",
"down_blocks.2.attentions.1.proj_in",
"down_blocks.2.attentions.1.proj_out",
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
"down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
"down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
"down_blocks.2.attentions.1.transformer_blocks.0.norm1",
"down_blocks.2.attentions.1.transformer_blocks.0.norm2",
"down_blocks.2.attentions.1.transformer_blocks.0.norm3",
"down_blocks.2.downsamplers.0.conv",
"down_blocks.2.resnets.0.conv1",
"down_blocks.2.resnets.0.conv2",
"down_blocks.2.resnets.0.conv_shortcut",
"down_blocks.2.resnets.0.norm1",
"down_blocks.2.resnets.0.norm2",
"down_blocks.2.resnets.0.time_emb_proj",
"down_blocks.2.resnets.1.conv1",
"down_blocks.2.resnets.1.conv2",
"down_blocks.2.resnets.1.norm1",
"down_blocks.2.resnets.1.norm2",
"down_blocks.2.resnets.1.time_emb_proj",
"down_blocks.3.resnets.0.conv1",
"down_blocks.3.resnets.0.conv2",
"down_blocks.3.resnets.0.norm1",
"down_blocks.3.resnets.0.norm2",
"down_blocks.3.resnets.0.time_emb_proj",
"down_blocks.3.resnets.1.conv1",
"down_blocks.3.resnets.1.conv2",
"down_blocks.3.resnets.1.norm1",
"down_blocks.3.resnets.1.norm2",
"down_blocks.3.resnets.1.time_emb_proj",
"mid_block.attentions.0.norm",
"mid_block.attentions.0.proj_in",
"mid_block.attentions.0.proj_out",
"mid_block.attentions.0.transformer_blocks.0.attn1.to_k",
"mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0",
"mid_block.attentions.0.transformer_blocks.0.attn1.to_q",
"mid_block.attentions.0.transformer_blocks.0.attn1.to_v",
"mid_block.attentions.0.transformer_blocks.0.attn2.to_k",
"mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0",
"mid_block.attentions.0.transformer_blocks.0.attn2.to_q",
"mid_block.attentions.0.transformer_blocks.0.attn2.to_v",
"mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj",
"mid_block.attentions.0.transformer_blocks.0.ff.net.2",
"mid_block.attentions.0.transformer_blocks.0.norm1",
"mid_block.attentions.0.transformer_blocks.0.norm2",
"mid_block.attentions.0.transformer_blocks.0.norm3",
"mid_block.resnets.0.conv1",
"mid_block.resnets.0.conv2",
"mid_block.resnets.0.norm1",
"mid_block.resnets.0.norm2",
"mid_block.resnets.0.time_emb_proj",
"mid_block.resnets.1.conv1",
"mid_block.resnets.1.conv2",
"mid_block.resnets.1.norm1",
"mid_block.resnets.1.norm2",
"mid_block.resnets.1.time_emb_proj",
"time_embedding.linear_1",
"time_embedding.linear_2",
"up_blocks.0.resnets.0.conv1",
"up_blocks.0.resnets.0.conv2",
"up_blocks.0.resnets.0.conv_shortcut",
"up_blocks.0.resnets.0.norm1",
"up_blocks.0.resnets.0.norm2",
"up_blocks.0.resnets.0.time_emb_proj",
"up_blocks.0.resnets.1.conv1",
"up_blocks.0.resnets.1.conv2",
"up_blocks.0.resnets.1.conv_shortcut",
"up_blocks.0.resnets.1.norm1",
"up_blocks.0.resnets.1.norm2",
"up_blocks.0.resnets.1.time_emb_proj",
"up_blocks.0.resnets.2.conv1",
"up_blocks.0.resnets.2.conv2",
"up_blocks.0.resnets.2.conv_shortcut",
"up_blocks.0.resnets.2.norm1",
"up_blocks.0.resnets.2.norm2",
"up_blocks.0.resnets.2.time_emb_proj",
"up_blocks.0.upsamplers.0.conv",
"up_blocks.1.attentions.0.norm",
"up_blocks.1.attentions.0.proj_in",
"up_blocks.1.attentions.0.proj_out",
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k",
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0",
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q",
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v",
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k",
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0",
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q",
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v",
"up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj",
"up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2",
"up_blocks.1.attentions.0.transformer_blocks.0.norm1",
"up_blocks.1.attentions.0.transformer_blocks.0.norm2",
"up_blocks.1.attentions.0.transformer_blocks.0.norm3",
"up_blocks.1.attentions.1.norm",
"up_blocks.1.attentions.1.proj_in",
"up_blocks.1.attentions.1.proj_out",
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k",
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0",
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q",
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v",
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k",
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0",
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q",
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v",
"up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj",
"up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2",
"up_blocks.1.attentions.1.transformer_blocks.0.norm1",
"up_blocks.1.attentions.1.transformer_blocks.0.norm2",
"up_blocks.1.attentions.1.transformer_blocks.0.norm3",
"up_blocks.1.attentions.2.norm",
"up_blocks.1.attentions.2.proj_in",
"up_blocks.1.attentions.2.proj_out",
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k",
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0",
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q",
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v",
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k",
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0",
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q",
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v",
"up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj",
"up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2",
"up_blocks.1.attentions.2.transformer_blocks.0.norm1",
"up_blocks.1.attentions.2.transformer_blocks.0.norm2",
"up_blocks.1.attentions.2.transformer_blocks.0.norm3",
"up_blocks.1.resnets.0.conv1",
"up_blocks.1.resnets.0.conv2",
"up_blocks.1.resnets.0.conv_shortcut",
"up_blocks.1.resnets.0.norm1",
"up_blocks.1.resnets.0.norm2",
"up_blocks.1.resnets.0.time_emb_proj",
"up_blocks.1.resnets.1.conv1",
"up_blocks.1.resnets.1.conv2",
"up_blocks.1.resnets.1.conv_shortcut",
"up_blocks.1.resnets.1.norm1",
"up_blocks.1.resnets.1.norm2",
"up_blocks.1.resnets.1.time_emb_proj",
"up_blocks.1.resnets.2.conv1",
"up_blocks.1.resnets.2.conv2",
"up_blocks.1.resnets.2.conv_shortcut",
"up_blocks.1.resnets.2.norm1",
"up_blocks.1.resnets.2.norm2",
"up_blocks.1.resnets.2.time_emb_proj",
"up_blocks.1.upsamplers.0.conv",
"up_blocks.2.attentions.0.norm",
"up_blocks.2.attentions.0.proj_in",
"up_blocks.2.attentions.0.proj_out",
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k",
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0",
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q",
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v",
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k",
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0",
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q",
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v",
"up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj",
"up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2",
"up_blocks.2.attentions.0.transformer_blocks.0.norm1",
"up_blocks.2.attentions.0.transformer_blocks.0.norm2",
"up_blocks.2.attentions.0.transformer_blocks.0.norm3",
"up_blocks.2.attentions.1.norm",
"up_blocks.2.attentions.1.proj_in",
"up_blocks.2.attentions.1.proj_out",
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k",
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0",
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q",
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v",
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k",
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0",
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q",
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v",
"up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj",
"up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2",
"up_blocks.2.attentions.1.transformer_blocks.0.norm1",
"up_blocks.2.attentions.1.transformer_blocks.0.norm2",
"up_blocks.2.attentions.1.transformer_blocks.0.norm3",
"up_blocks.2.attentions.2.norm",
"up_blocks.2.attentions.2.proj_in",
"up_blocks.2.attentions.2.proj_out",
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k",
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0",
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q",
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v",
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k",
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0",
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q",
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v",
"up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj",
"up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2",
"up_blocks.2.attentions.2.transformer_blocks.0.norm1",
"up_blocks.2.attentions.2.transformer_blocks.0.norm2",
"up_blocks.2.attentions.2.transformer_blocks.0.norm3",
"up_blocks.2.resnets.0.conv1",
"up_blocks.2.resnets.0.conv2",
"up_blocks.2.resnets.0.conv_shortcut",
"up_blocks.2.resnets.0.norm1",
"up_blocks.2.resnets.0.norm2",
"up_blocks.2.resnets.0.time_emb_proj",
"up_blocks.2.resnets.1.conv1",
"up_blocks.2.resnets.1.conv2",
"up_blocks.2.resnets.1.conv_shortcut",
"up_blocks.2.resnets.1.norm1",
"up_blocks.2.resnets.1.norm2",
"up_blocks.2.resnets.1.time_emb_proj",
"up_blocks.2.resnets.2.conv1",
"up_blocks.2.resnets.2.conv2",
"up_blocks.2.resnets.2.conv_shortcut",
"up_blocks.2.resnets.2.norm1",
"up_blocks.2.resnets.2.norm2",
"up_blocks.2.resnets.2.time_emb_proj",
"up_blocks.2.upsamplers.0.conv",
"up_blocks.3.attentions.0.norm",
"up_blocks.3.attentions.0.proj_in",
"up_blocks.3.attentions.0.proj_out",
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k",
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0",
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q",
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v",
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k",
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0",
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q",
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v",
"up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj",
"up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2",
"up_blocks.3.attentions.0.transformer_blocks.0.norm1",
"up_blocks.3.attentions.0.transformer_blocks.0.norm2",
"up_blocks.3.attentions.0.transformer_blocks.0.norm3",
"up_blocks.3.attentions.1.norm",
"up_blocks.3.attentions.1.proj_in",
"up_blocks.3.attentions.1.proj_out",
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k",
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0",
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q",
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v",
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k",
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0",
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q",
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v",
"up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj",
"up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2",
"up_blocks.3.attentions.1.transformer_blocks.0.norm1",
"up_blocks.3.attentions.1.transformer_blocks.0.norm2",
"up_blocks.3.attentions.1.transformer_blocks.0.norm3",
"up_blocks.3.attentions.2.norm",
"up_blocks.3.attentions.2.proj_in",
"up_blocks.3.attentions.2.proj_out",
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k",
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0",
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q",
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v",
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k",
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0",
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q",
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v",
"up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj",
"up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2",
"up_blocks.3.attentions.2.transformer_blocks.0.norm1",
"up_blocks.3.attentions.2.transformer_blocks.0.norm2",
"up_blocks.3.attentions.2.transformer_blocks.0.norm3",
"up_blocks.3.resnets.0.conv1",
"up_blocks.3.resnets.0.conv2",
"up_blocks.3.resnets.0.conv_shortcut",
"up_blocks.3.resnets.0.norm1",
"up_blocks.3.resnets.0.norm2",
"up_blocks.3.resnets.0.time_emb_proj",
"up_blocks.3.resnets.1.conv1",
"up_blocks.3.resnets.1.conv2",
"up_blocks.3.resnets.1.conv_shortcut",
"up_blocks.3.resnets.1.norm1",
"up_blocks.3.resnets.1.norm2",
"up_blocks.3.resnets.1.time_emb_proj",
"up_blocks.3.resnets.2.conv1",
"up_blocks.3.resnets.2.conv2",
"up_blocks.3.resnets.2.conv_shortcut",
"up_blocks.3.resnets.2.norm1",
"up_blocks.3.resnets.2.norm2",
"up_blocks.3.resnets.2.time_emb_proj"
]