Can't use VAE #1375
Replies: 3 comments 8 replies
-
try removing the argument --no-half-vae in the last cell |
Beta Was this translation helpful? Give feedback.
-
No dice :(
|
Beta Was this translation helpful? Give feedback.
-
Is this what you meant? Just checking if I got it right.
|
Beta Was this translation helpful? Give feedback.
Uh oh!
There was an error while loading. Please reload this page.
-
Hi,
I'm using Google Colab with models on my Google Drive.
I was having problems loading VAE checkpoints, so I decided to start from scratch. I renamed my old sd folder, copied and ran the latest version, waited for everything to reinstall… and the problem is still there.
When trying to load VAE (it doesn't matter what VAE or which model I am using) I get the following error:
Loading VAE weights from: /content/gdrive/MyDrive/sd/stable-diffusion-webui/models/VAE/anything-v3-vae-swapped.ckpt changing setting sd_vae to anything-v3-vae-swapped: RuntimeError Traceback (most recent call last): File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/shared.py", line 515, in set self.data_labels[key].onchange() File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/call_queue.py", line 15, in f res = func(*args, **kwargs) File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/webui.py", line 82, in <lambda> shared.opts.onchange("sd_vae", wrap_queued_call(lambda: modules.sd_vae.reload_vae_weights()), call=False) File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_vae.py", line 234, in reload_vae_weights load_vae(sd_model, vae_file) File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_vae.py", line 178, in load_vae _load_vae_dict(model, vae_dict_1) File "/content/gdrive/MyDrive/sd/stable-diffusion-webui/modules/sd_vae.py", line 205, in _load_vae_dict model.first_stage_model.load_state_dict(vae_dict_1) File "/usr/local/lib/python3.8/dist-packages/torch/nn/modules/module.py", line 1667, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format( RuntimeError: Error(s) in loading state_dict for AutoencoderKL: Missing key(s) in state_dict: "encoder.conv_in.weight", "encoder.conv_in.bias", "encoder.down.0.block.0.norm1.weight", "encoder.down.0.block.0.norm1.bias", "encoder.down.0.block.0.conv1.weight", "encoder.down.0.block.0.conv1.bias", "encoder.down.0.block.0.norm2.weight", "encoder.down.0.block.0.norm2.bias", "encoder.down.0.block.0.conv2.weight", "encoder.down.0.block.0.conv2.bias", "encoder.down.0.block.1.norm1.weight", "encoder.down.0.block.1.norm1.bias", "encoder.down.0.block.1.conv1.weight", "encoder.down.0.block.1.conv1.bias", "encoder.down.0.block.1.norm2.weight", "encoder.down.0.block.1.norm2.bias", "encoder.down.0.block.1.conv2.weight", "encoder.down.0.block.1.conv2.bias", "encoder.down.0.downsample.conv.weight", "encoder.down.0.downsample.conv.bias", "encoder.down.1.block.0.norm1.weight", "encoder.down.1.block.0.norm1.bias", "encoder.down.1.block.0.conv1.weight", "encoder.down.1.block.0.conv1.bias", "encoder.down.1.block.0.norm2.weight", "encoder.down.1.block.0.norm2.bias", "encoder.down.1.block.0.conv2.weight", "encoder.down.1.block.0.conv2.bias", "encoder.down.1.block.0.nin_shortcut.weight", "encoder.down.1.block.0.nin_shortcut.bias", "encoder.down.1.block.1.norm1.weight", "encoder.down.1.block.1.norm1.bias", "encoder.down.1.block.1.conv1.weight", "encoder.down.1.block.1.conv1.bias", "encoder.down.1.block.1.norm2.weight", "encoder.down.1.block.1.norm2.bias", "encoder.down.1.block.1.conv2.weight", "encoder.down.1.block.1.conv2.bias", "encoder.down.1.downsample.conv.weight", "encoder.down.1.downsample.conv.bias", "encoder.down.2.block.0.norm1.weight", "encoder.down.2.block.0.norm1.bias", "encoder.down.2.block.0.conv1.weight", "encoder.down.2.block.0.conv1.bias", "encoder.down.2.block.0.norm2.weight", "encoder.down.2.block.0.norm2.bias", "encoder.down.2.block.0.conv2.weight", "encoder.down.2.block.0.conv2.bias", "encoder.down.2.block.0.nin_shortcut.weight", "encoder.down.2.block.0.nin_shortcut.bias", "encoder.down.2.block.1.norm1.weight", "encoder.down.2.block.1.norm1.bias", "encoder.down.2.block.1.conv1.weight", "encoder.down.2.block.1.conv1.bias", "encoder.down.2.block.1.norm2.weight", "encoder.down.2.block.1.norm2.bias", "encoder.down.2.block.1.conv2.weight", "encoder.down.2.block.1.conv2.bias", "encoder.down.2.downsample.conv.weight", "encoder.down.2.downsample.conv.bias", "encoder.down.3.block.0.norm1.weight", "encoder.down.3.block.0.norm1.bias", "encoder.down.3.block.0.conv1.weight", "encoder.down.3.block.0.conv1.bias", "encoder.down.3.block.0.norm2.weight", "encoder.down.3.block.0.norm2.bias", "encoder.down.3.block.0.conv2.weight", "encoder.down.3.block.0.conv2.bias", "encoder.down.3.block.1.norm1.weight", "encoder.down.3.block.1.norm1.bias", "encoder.down.3.block.1.conv1.weight", "encoder.down.3.block.1.conv1.bias", "encoder.down.3.block.1.norm2.weight", "encoder.down.3.block.1.norm2.bias", "encoder.down.3.block.1.conv2.weight", "encoder.down.3.block.1.conv2.bias", "encoder.mid.block_1.norm1.weight", "encoder.mid.block_1.norm1.bias", "encoder.mid.block_1.conv1.weight", "encoder.mid.block_1.conv1.bias", "encoder.mid.block_1.norm2.weight", "encoder.mid.block_1.norm2.bias", "encoder.mid.block_1.conv2.weight", "encoder.mid.block_1.conv2.bias", "encoder.mid.attn_1.norm.weight", "encoder.mid.attn_1.norm.bias", "encoder.mid.attn_1.q.weight", "encoder.mid.attn_1.q.bias", "encoder.mid.attn_1.k.weight", "encoder.mid.attn_1.k.bias", "encoder.mid.attn_1.v.weight", "encoder.mid.attn_1.v.bias", "encoder.mid.attn_1.proj_out.weight", "encoder.mid.attn_1.proj_out.bias", "encoder.mid.block_2.norm1.weight", "encoder.mid.block_2.norm1.bias", "encoder.mid.block_2.conv1.weight", "encoder.mid.block_2.conv1.bias", "encoder.mid.block_2.norm2.weight", "encoder.mid.block_2.norm2.bias", "encoder.mid.block_2.conv2.weight", "encoder.mid.block_2.conv2.bias", "encoder.norm_out.weight", "encoder.norm_out.bias", "encoder.conv_out.weight", "encoder.conv_out.bias", "decoder.conv_in.weight", "decoder.conv_in.bias", "decoder.mid.block_1.norm1.weight", "decoder.mid.block_1.norm1.bias", "decoder.mid.block_1.conv1.weight", "decoder.mid.block_1.conv1.bias", "decoder.mid.block_1.norm2.weight", "decoder.mid.block_1.norm2.bias", "decoder.mid.block_1.conv2.weight", "decoder.mid.block_1.conv2.bias", "decoder.mid.attn_1.norm.weight", "decoder.mid.attn_1.norm.bias", "decoder.mid.attn_1.q.weight", "decoder.mid.attn_1.q.bias", "decoder.mid.attn_1.k.weight", "decoder.mid.attn_1.k.bias", "decoder.mid.attn_1.v.weight", "decoder.mid.attn_1.v.bias", "decoder.mid.attn_1.proj_out.weight", "decoder.mid.attn_1.proj_out.bias", "decoder.mid.block_2.norm1.weight", "decoder.mid.block_2.norm1.bias", "decoder.mid.block_2.conv1.weight", "decoder.mid.block_2.conv1.bias", "decoder.mid.block_2.norm2.weight", "decoder.mid.block_2.norm2.bias", "decoder.mid.block_2.conv2.weight", "decoder.mid.block_2.conv2.bias", "decoder.up.0.block.0.norm1.weight", "decoder.up.0.block.0.norm1.bias", "decoder.up.0.block.0.conv1.weight", "decoder.up.0.block.0.conv1.bias", "decoder.up.0.block.0.norm2.weight", "decoder.up.0.block.0.norm2.bias", "decoder.up.0.block.0.conv2.weight", "decoder.up.0.block.0.conv2.bias", "decoder.up.0.block.0.nin_shortcut.weight", "decoder.up.0.block.0.nin_shortcut.bias", "decoder.up.0.block.1.norm1.weight", "decoder.up.0.block.1.norm1.bias", "decoder.up.0.block.1.conv1.weight", "decoder.up.0.block.1.conv1.bias", "decoder.up.0.block.1.norm2.weight", "decoder.up.0.block.1.norm2.bias", "decoder.up.0.block.1.conv2.weight", "decoder.up.0.block.1.conv2.bias", "decoder.up.0.block.2.norm1.weight", "decoder.up.0.block.2.norm1.bias", "decoder.up.0.block.2.conv1.weight", "decoder.up.0.block.2.conv1.bias", "decoder.up.0.block.2.norm2.weight", "decoder.up.0.block.2.norm2.bias", "decoder.up.0.block.2.conv2.weight", "decoder.up.0.block.2.conv2.bias", "decoder.up.1.block.0.norm1.weight", "decoder.up.1.block.0.norm1.bias", "decoder.up.1.block.0.conv1.weight", "decoder.up.1.block.0.conv1.bias", "decoder.up.1.block.0.norm2.weight", "decoder.up.1.block.0.norm2.bias", "decoder.up.1.block.0.conv2.weight", "decoder.up.1.block.0.conv2.bias", "decoder.up.1.block.0.nin_shortcut.weight", "decoder.up.1.block.0.nin_shortcut.bias", "decoder.up.1.block.1.norm1.weight", "decoder.up.1.block.1.norm1.bias", "decoder.up.1.block.1.conv1.weight", "decoder.up.1.block.1.conv1.bias", "decoder.up.1.block.1.norm2.weight", "decoder.up.1.block.1.norm2.bias", "decoder.up.1.block.1.conv2.weight", "decoder.up.1.block.1.conv2.bias", "decoder.up.1.block.2.norm1.weight", "decoder.up.1.block.2.norm1.bias", "decoder.up.1.block.2.conv1.weight", "decoder.up.1.block.2.conv1.bias", "decoder.up.1.block.2.norm2.weight", "decoder.up.1.block.2.norm2.bias", "decoder.up.1.block.2.conv2.weight", "decoder.up.1.block.2.conv2.bias", "decoder.up.1.upsample.conv.weight", "decoder.up.1.upsample.conv.bias", "decoder.up.2.block.0.norm1.weight", "decoder.up.2.block.0.norm1.bias", "decoder.up.2.block.0.conv1.weight", "decoder.up.2.block.0.conv1.bias", "decoder.up.2.block.0.norm2.weight", "decoder.up.2.block.0.norm2.bias", "decoder.up.2.block.0.conv2.weight", "decoder.up.2.block.0.conv2.bias", "decoder.up.2.block.1.norm1.weight", "decoder.up.2.block.1.norm1.bias", "decoder.up.2.block.1.conv1.weight", "decoder.up.2.block.1.conv1.bias", "decoder.up.2.block.1.norm2.weight", "decoder.up.2.block.1.norm2.bias", "decoder.up.2.block.1.conv2.weight", "decoder.up.2.block.1.conv2.bias", "decoder.up.2.block.2.norm1.weight", "decoder.up.2.block.2.norm1.bias", "decoder.up.2.block.2.conv1.weight", "decoder.up.2.block.2.conv1.bias", "decoder.up.2.block.2.norm2.weight", "decoder.up.2.block.2.norm2.bias", "decoder.up.2.block.2.conv2.weight", "decoder.up.2.block.2.conv2.bias", "decoder.up.2.upsample.conv.weight", "decoder.up.2.upsample.conv.bias", "decoder.up.3.block.0.norm1.weight", "decoder.up.3.block.0.norm1.bias", "decoder.up.3.block.0.conv1.weight", "decoder.up.3.block.0.conv1.bias", "decoder.up.3.block.0.norm2.weight", "decoder.up.3.block.0.norm2.bias", "decoder.up.3.block.0.conv2.weight", "decoder.up.3.block.0.conv2.bias", "decoder.up.3.block.1.norm1.weight", "decoder.up.3.block.1.norm1.bias", "decoder.up.3.block.1.conv1.weight", "decoder.up.3.block.1.conv1.bias", "decoder.up.3.block.1.norm2.weight", "decoder.up.3.block.1.norm2.bias", "decoder.up.3.block.1.conv2.weight", "decoder.up.3.block.1.conv2.bias", "decoder.up.3.block.2.norm1.weight", "decoder.up.3.block.2.norm1.bias", "decoder.up.3.block.2.conv1.weight", "decoder.up.3.block.2.conv1.bias", "decoder.up.3.block.2.norm2.weight", "decoder.up.3.block.2.norm2.bias", "decoder.up.3.block.2.conv2.weight", "decoder.up.3.block.2.conv2.bias", "decoder.up.3.upsample.conv.weight", "decoder.up.3.upsample.conv.bias", "decoder.norm_out.weight", "decoder.norm_out.bias", "decoder.conv_out.weight", "decoder.conv_out.bias", "quant_conv.weight", "quant_conv.bias", "post_quant_conv.weight", "post_quant_conv.bias". Unexpected key(s) in state_dict: "model.diffusion_model.input_blocks.0.0.weight", "model.diffusion_model.input_blocks.0.0.bias", "model.diffusion_model.time_embed.0.weight", "model.diffusion_model.time_embed.0.bias", "model.diffusion_model.time_embed.2.weight", "model.diffusion_model.time_embed.2.bias", "model.diffusion_model.input_blocks.1.1.norm.weight", "model.diffusion_model.input_blocks.1.1.norm.bias", "model.diffusion_model.input_blocks.1.1.proj_in.weight", "model.diffusion_model.input_blocks.1.1.proj_in.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.input_blocks.1.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.input_blocks.1.1.proj_out.weight", "model.diffusion_model.input_blocks.1.1.proj_out.bias", "model.diffusion_model.input_blocks.2.1.norm.weight", "model.diffusion_model.input_blocks.2.1.norm.bias", "model.diffusion_model.input_blocks.2.1.proj_in.weight", "model.diffusion_model.input_blocks.2.1.proj_in.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.input_blocks.2.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.input_blocks.2.1.proj_out.weight", "model.diffusion_model.input_blocks.2.1.proj_out.bias", "model.diffusion_model.input_blocks.1.0.in_layers.0.weight", "model.diffusion_model.input_blocks.1.0.in_layers.0.bias", "model.diffusion_model.input_blocks.1.0.in_layers.2.weight", "model.diffusion_model.input_blocks.1.0.in_layers.2.bias", "model.diffusion_model.input_blocks.1.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.1.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.1.0.out_layers.0.weight", "model.diffusion_model.input_blocks.1.0.out_layers.0.bias", "model.diffusion_model.input_blocks.1.0.out_layers.3.weight", "model.diffusion_model.input_blocks.1.0.out_layers.3.bias", "model.diffusion_model.input_blocks.2.0.in_layers.0.weight", "model.diffusion_model.input_blocks.2.0.in_layers.0.bias", "model.diffusion_model.input_blocks.2.0.in_layers.2.weight", "model.diffusion_model.input_blocks.2.0.in_layers.2.bias", "model.diffusion_model.input_blocks.2.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.2.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.2.0.out_layers.0.weight", "model.diffusion_model.input_blocks.2.0.out_layers.0.bias", "model.diffusion_model.input_blocks.2.0.out_layers.3.weight", "model.diffusion_model.input_blocks.2.0.out_layers.3.bias", "model.diffusion_model.input_blocks.3.0.op.weight", "model.diffusion_model.input_blocks.3.0.op.bias", "model.diffusion_model.input_blocks.4.1.norm.weight", "model.diffusion_model.input_blocks.4.1.norm.bias", "model.diffusion_model.input_blocks.4.1.proj_in.weight", "model.diffusion_model.input_blocks.4.1.proj_in.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.input_blocks.4.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.input_blocks.4.1.proj_out.weight", "model.diffusion_model.input_blocks.4.1.proj_out.bias", "model.diffusion_model.input_blocks.5.1.norm.weight", "model.diffusion_model.input_blocks.5.1.norm.bias", "model.diffusion_model.input_blocks.5.1.proj_in.weight", "model.diffusion_model.input_blocks.5.1.proj_in.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.input_blocks.5.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.input_blocks.5.1.proj_out.weight", "model.diffusion_model.input_blocks.5.1.proj_out.bias", "model.diffusion_model.input_blocks.4.0.in_layers.0.weight", "model.diffusion_model.input_blocks.4.0.in_layers.0.bias", "model.diffusion_model.input_blocks.4.0.in_layers.2.weight", "model.diffusion_model.input_blocks.4.0.in_layers.2.bias", "model.diffusion_model.input_blocks.4.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.4.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.4.0.out_layers.0.weight", "model.diffusion_model.input_blocks.4.0.out_layers.0.bias", "model.diffusion_model.input_blocks.4.0.out_layers.3.weight", "model.diffusion_model.input_blocks.4.0.out_layers.3.bias", "model.diffusion_model.input_blocks.4.0.skip_connection.weight", "model.diffusion_model.input_blocks.4.0.skip_connection.bias", "model.diffusion_model.input_blocks.5.0.in_layers.0.weight", "model.diffusion_model.input_blocks.5.0.in_layers.0.bias", "model.diffusion_model.input_blocks.5.0.in_layers.2.weight", "model.diffusion_model.input_blocks.5.0.in_layers.2.bias", "model.diffusion_model.input_blocks.5.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.5.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.5.0.out_layers.0.weight", "model.diffusion_model.input_blocks.5.0.out_layers.0.bias", "model.diffusion_model.input_blocks.5.0.out_layers.3.weight", "model.diffusion_model.input_blocks.5.0.out_layers.3.bias", "model.diffusion_model.input_blocks.6.0.op.weight", "model.diffusion_model.input_blocks.6.0.op.bias", "model.diffusion_model.input_blocks.7.1.norm.weight", "model.diffusion_model.input_blocks.7.1.norm.bias", "model.diffusion_model.input_blocks.7.1.proj_in.weight", "model.diffusion_model.input_blocks.7.1.proj_in.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.input_blocks.7.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.input_blocks.7.1.proj_out.weight", "model.diffusion_model.input_blocks.7.1.proj_out.bias", "model.diffusion_model.input_blocks.8.1.norm.weight", "model.diffusion_model.input_blocks.8.1.norm.bias", "model.diffusion_model.input_blocks.8.1.proj_in.weight", "model.diffusion_model.input_blocks.8.1.proj_in.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.input_blocks.8.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.input_blocks.8.1.proj_out.weight", "model.diffusion_model.input_blocks.8.1.proj_out.bias", "model.diffusion_model.input_blocks.7.0.in_layers.0.weight", "model.diffusion_model.input_blocks.7.0.in_layers.0.bias", "model.diffusion_model.input_blocks.7.0.in_layers.2.weight", "model.diffusion_model.input_blocks.7.0.in_layers.2.bias", "model.diffusion_model.input_blocks.7.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.7.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.7.0.out_layers.0.weight", "model.diffusion_model.input_blocks.7.0.out_layers.0.bias", "model.diffusion_model.input_blocks.7.0.out_layers.3.weight", "model.diffusion_model.input_blocks.7.0.out_layers.3.bias", "model.diffusion_model.input_blocks.7.0.skip_connection.weight", "model.diffusion_model.input_blocks.7.0.skip_connection.bias", "model.diffusion_model.input_blocks.8.0.in_layers.0.weight", "model.diffusion_model.input_blocks.8.0.in_layers.0.bias", "model.diffusion_model.input_blocks.8.0.in_layers.2.weight", "model.diffusion_model.input_blocks.8.0.in_layers.2.bias", "model.diffusion_model.input_blocks.8.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.8.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.8.0.out_layers.0.weight", "model.diffusion_model.input_blocks.8.0.out_layers.0.bias", "model.diffusion_model.input_blocks.8.0.out_layers.3.weight", "model.diffusion_model.input_blocks.8.0.out_layers.3.bias", "model.diffusion_model.input_blocks.9.0.op.weight", "model.diffusion_model.input_blocks.9.0.op.bias", "model.diffusion_model.input_blocks.10.0.in_layers.0.weight", "model.diffusion_model.input_blocks.10.0.in_layers.0.bias", "model.diffusion_model.input_blocks.10.0.in_layers.2.weight", "model.diffusion_model.input_blocks.10.0.in_layers.2.bias", "model.diffusion_model.input_blocks.10.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.10.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.10.0.out_layers.0.weight", "model.diffusion_model.input_blocks.10.0.out_layers.0.bias", "model.diffusion_model.input_blocks.10.0.out_layers.3.weight", "model.diffusion_model.input_blocks.10.0.out_layers.3.bias", "model.diffusion_model.input_blocks.11.0.in_layers.0.weight", "model.diffusion_model.input_blocks.11.0.in_layers.0.bias", "model.diffusion_model.input_blocks.11.0.in_layers.2.weight", "model.diffusion_model.input_blocks.11.0.in_layers.2.bias", "model.diffusion_model.input_blocks.11.0.emb_layers.1.weight", "model.diffusion_model.input_blocks.11.0.emb_layers.1.bias", "model.diffusion_model.input_blocks.11.0.out_layers.0.weight", "model.diffusion_model.input_blocks.11.0.out_layers.0.bias", "model.diffusion_model.input_blocks.11.0.out_layers.3.weight", "model.diffusion_model.input_blocks.11.0.out_layers.3.bias", "model.diffusion_model.output_blocks.0.0.in_layers.0.weight", "model.diffusion_model.output_blocks.0.0.in_layers.0.bias", "model.diffusion_model.output_blocks.0.0.in_layers.2.weight", "model.diffusion_model.output_blocks.0.0.in_layers.2.bias", "model.diffusion_model.output_blocks.0.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.0.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.0.0.out_layers.0.weight", "model.diffusion_model.output_blocks.0.0.out_layers.0.bias", "model.diffusion_model.output_blocks.0.0.out_layers.3.weight", "model.diffusion_model.output_blocks.0.0.out_layers.3.bias", "model.diffusion_model.output_blocks.0.0.skip_connection.weight", "model.diffusion_model.output_blocks.0.0.skip_connection.bias", "model.diffusion_model.output_blocks.1.0.in_layers.0.weight", "model.diffusion_model.output_blocks.1.0.in_layers.0.bias", "model.diffusion_model.output_blocks.1.0.in_layers.2.weight", "model.diffusion_model.output_blocks.1.0.in_layers.2.bias", "model.diffusion_model.output_blocks.1.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.1.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.1.0.out_layers.0.weight", "model.diffusion_model.output_blocks.1.0.out_layers.0.bias", "model.diffusion_model.output_blocks.1.0.out_layers.3.weight", "model.diffusion_model.output_blocks.1.0.out_layers.3.bias", "model.diffusion_model.output_blocks.1.0.skip_connection.weight", "model.diffusion_model.output_blocks.1.0.skip_connection.bias", "model.diffusion_model.output_blocks.2.0.in_layers.0.weight", "model.diffusion_model.output_blocks.2.0.in_layers.0.bias", "model.diffusion_model.output_blocks.2.0.in_layers.2.weight", "model.diffusion_model.output_blocks.2.0.in_layers.2.bias", "model.diffusion_model.output_blocks.2.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.2.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.2.0.out_layers.0.weight", "model.diffusion_model.output_blocks.2.0.out_layers.0.bias", "model.diffusion_model.output_blocks.2.0.out_layers.3.weight", "model.diffusion_model.output_blocks.2.0.out_layers.3.bias", "model.diffusion_model.output_blocks.2.0.skip_connection.weight", "model.diffusion_model.output_blocks.2.0.skip_connection.bias", "model.diffusion_model.output_blocks.2.1.conv.weight", "model.diffusion_model.output_blocks.2.1.conv.bias", "model.diffusion_model.output_blocks.3.1.norm.weight", "model.diffusion_model.output_blocks.3.1.norm.bias", "model.diffusion_model.output_blocks.3.1.proj_in.weight", "model.diffusion_model.output_blocks.3.1.proj_in.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.3.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.3.1.proj_out.weight", "model.diffusion_model.output_blocks.3.1.proj_out.bias", "model.diffusion_model.output_blocks.4.1.norm.weight", "model.diffusion_model.output_blocks.4.1.norm.bias", "model.diffusion_model.output_blocks.4.1.proj_in.weight", "model.diffusion_model.output_blocks.4.1.proj_in.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.4.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.4.1.proj_out.weight", "model.diffusion_model.output_blocks.4.1.proj_out.bias", "model.diffusion_model.output_blocks.5.1.norm.weight", "model.diffusion_model.output_blocks.5.1.norm.bias", "model.diffusion_model.output_blocks.5.1.proj_in.weight", "model.diffusion_model.output_blocks.5.1.proj_in.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.5.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.5.1.proj_out.weight", "model.diffusion_model.output_blocks.5.1.proj_out.bias", "model.diffusion_model.output_blocks.3.0.in_layers.0.weight", "model.diffusion_model.output_blocks.3.0.in_layers.0.bias", "model.diffusion_model.output_blocks.3.0.in_layers.2.weight", "model.diffusion_model.output_blocks.3.0.in_layers.2.bias", "model.diffusion_model.output_blocks.3.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.3.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.3.0.out_layers.0.weight", "model.diffusion_model.output_blocks.3.0.out_layers.0.bias", "model.diffusion_model.output_blocks.3.0.out_layers.3.weight", "model.diffusion_model.output_blocks.3.0.out_layers.3.bias", "model.diffusion_model.output_blocks.3.0.skip_connection.weight", "model.diffusion_model.output_blocks.3.0.skip_connection.bias", "model.diffusion_model.output_blocks.4.0.in_layers.0.weight", "model.diffusion_model.output_blocks.4.0.in_layers.0.bias", "model.diffusion_model.output_blocks.4.0.in_layers.2.weight", "model.diffusion_model.output_blocks.4.0.in_layers.2.bias", "model.diffusion_model.output_blocks.4.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.4.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.4.0.out_layers.0.weight", "model.diffusion_model.output_blocks.4.0.out_layers.0.bias", "model.diffusion_model.output_blocks.4.0.out_layers.3.weight", "model.diffusion_model.output_blocks.4.0.out_layers.3.bias", "model.diffusion_model.output_blocks.4.0.skip_connection.weight", "model.diffusion_model.output_blocks.4.0.skip_connection.bias", "model.diffusion_model.output_blocks.5.0.in_layers.0.weight", "model.diffusion_model.output_blocks.5.0.in_layers.0.bias", "model.diffusion_model.output_blocks.5.0.in_layers.2.weight", "model.diffusion_model.output_blocks.5.0.in_layers.2.bias", "model.diffusion_model.output_blocks.5.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.5.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.5.0.out_layers.0.weight", "model.diffusion_model.output_blocks.5.0.out_layers.0.bias", "model.diffusion_model.output_blocks.5.0.out_layers.3.weight", "model.diffusion_model.output_blocks.5.0.out_layers.3.bias", "model.diffusion_model.output_blocks.5.0.skip_connection.weight", "model.diffusion_model.output_blocks.5.0.skip_connection.bias", "model.diffusion_model.output_blocks.5.2.conv.weight", "model.diffusion_model.output_blocks.5.2.conv.bias", "model.diffusion_model.output_blocks.6.1.norm.weight", "model.diffusion_model.output_blocks.6.1.norm.bias", "model.diffusion_model.output_blocks.6.1.proj_in.weight", "model.diffusion_model.output_blocks.6.1.proj_in.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.6.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.6.1.proj_out.weight", "model.diffusion_model.output_blocks.6.1.proj_out.bias", "model.diffusion_model.output_blocks.7.1.norm.weight", "model.diffusion_model.output_blocks.7.1.norm.bias", "model.diffusion_model.output_blocks.7.1.proj_in.weight", "model.diffusion_model.output_blocks.7.1.proj_in.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.7.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.7.1.proj_out.weight", "model.diffusion_model.output_blocks.7.1.proj_out.bias", "model.diffusion_model.output_blocks.8.1.norm.weight", "model.diffusion_model.output_blocks.8.1.norm.bias", "model.diffusion_model.output_blocks.8.1.proj_in.weight", "model.diffusion_model.output_blocks.8.1.proj_in.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.8.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.8.1.proj_out.weight", "model.diffusion_model.output_blocks.8.1.proj_out.bias", "model.diffusion_model.output_blocks.6.0.in_layers.0.weight", "model.diffusion_model.output_blocks.6.0.in_layers.0.bias", "model.diffusion_model.output_blocks.6.0.in_layers.2.weight", "model.diffusion_model.output_blocks.6.0.in_layers.2.bias", "model.diffusion_model.output_blocks.6.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.6.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.6.0.out_layers.0.weight", "model.diffusion_model.output_blocks.6.0.out_layers.0.bias", "model.diffusion_model.output_blocks.6.0.out_layers.3.weight", "model.diffusion_model.output_blocks.6.0.out_layers.3.bias", "model.diffusion_model.output_blocks.6.0.skip_connection.weight", "model.diffusion_model.output_blocks.6.0.skip_connection.bias", "model.diffusion_model.output_blocks.7.0.in_layers.0.weight", "model.diffusion_model.output_blocks.7.0.in_layers.0.bias", "model.diffusion_model.output_blocks.7.0.in_layers.2.weight", "model.diffusion_model.output_blocks.7.0.in_layers.2.bias", "model.diffusion_model.output_blocks.7.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.7.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.7.0.out_layers.0.weight", "model.diffusion_model.output_blocks.7.0.out_layers.0.bias", "model.diffusion_model.output_blocks.7.0.out_layers.3.weight", "model.diffusion_model.output_blocks.7.0.out_layers.3.bias", "model.diffusion_model.output_blocks.7.0.skip_connection.weight", "model.diffusion_model.output_blocks.7.0.skip_connection.bias", "model.diffusion_model.output_blocks.8.0.in_layers.0.weight", "model.diffusion_model.output_blocks.8.0.in_layers.0.bias", "model.diffusion_model.output_blocks.8.0.in_layers.2.weight", "model.diffusion_model.output_blocks.8.0.in_layers.2.bias", "model.diffusion_model.output_blocks.8.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.8.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.8.0.out_layers.0.weight", "model.diffusion_model.output_blocks.8.0.out_layers.0.bias", "model.diffusion_model.output_blocks.8.0.out_layers.3.weight", "model.diffusion_model.output_blocks.8.0.out_layers.3.bias", "model.diffusion_model.output_blocks.8.0.skip_connection.weight", "model.diffusion_model.output_blocks.8.0.skip_connection.bias", "model.diffusion_model.output_blocks.8.2.conv.weight", "model.diffusion_model.output_blocks.8.2.conv.bias", "model.diffusion_model.output_blocks.9.1.norm.weight", "model.diffusion_model.output_blocks.9.1.norm.bias", "model.diffusion_model.output_blocks.9.1.proj_in.weight", "model.diffusion_model.output_blocks.9.1.proj_in.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.9.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.9.1.proj_out.weight", "model.diffusion_model.output_blocks.9.1.proj_out.bias", "model.diffusion_model.output_blocks.10.1.norm.weight", "model.diffusion_model.output_blocks.10.1.norm.bias", "model.diffusion_model.output_blocks.10.1.proj_in.weight", "model.diffusion_model.output_blocks.10.1.proj_in.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.10.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.10.1.proj_out.weight", "model.diffusion_model.output_blocks.10.1.proj_out.bias", "model.diffusion_model.output_blocks.11.1.norm.weight", "model.diffusion_model.output_blocks.11.1.norm.bias", "model.diffusion_model.output_blocks.11.1.proj_in.weight", "model.diffusion_model.output_blocks.11.1.proj_in.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.output_blocks.11.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.output_blocks.11.1.proj_out.weight", "model.diffusion_model.output_blocks.11.1.proj_out.bias", "model.diffusion_model.output_blocks.9.0.in_layers.0.weight", "model.diffusion_model.output_blocks.9.0.in_layers.0.bias", "model.diffusion_model.output_blocks.9.0.in_layers.2.weight", "model.diffusion_model.output_blocks.9.0.in_layers.2.bias", "model.diffusion_model.output_blocks.9.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.9.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.9.0.out_layers.0.weight", "model.diffusion_model.output_blocks.9.0.out_layers.0.bias", "model.diffusion_model.output_blocks.9.0.out_layers.3.weight", "model.diffusion_model.output_blocks.9.0.out_layers.3.bias", "model.diffusion_model.output_blocks.9.0.skip_connection.weight", "model.diffusion_model.output_blocks.9.0.skip_connection.bias", "model.diffusion_model.output_blocks.10.0.in_layers.0.weight", "model.diffusion_model.output_blocks.10.0.in_layers.0.bias", "model.diffusion_model.output_blocks.10.0.in_layers.2.weight", "model.diffusion_model.output_blocks.10.0.in_layers.2.bias", "model.diffusion_model.output_blocks.10.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.10.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.10.0.out_layers.0.weight", "model.diffusion_model.output_blocks.10.0.out_layers.0.bias", "model.diffusion_model.output_blocks.10.0.out_layers.3.weight", "model.diffusion_model.output_blocks.10.0.out_layers.3.bias", "model.diffusion_model.output_blocks.10.0.skip_connection.weight", "model.diffusion_model.output_blocks.10.0.skip_connection.bias", "model.diffusion_model.output_blocks.11.0.in_layers.0.weight", "model.diffusion_model.output_blocks.11.0.in_layers.0.bias", "model.diffusion_model.output_blocks.11.0.in_layers.2.weight", "model.diffusion_model.output_blocks.11.0.in_layers.2.bias", "model.diffusion_model.output_blocks.11.0.emb_layers.1.weight", "model.diffusion_model.output_blocks.11.0.emb_layers.1.bias", "model.diffusion_model.output_blocks.11.0.out_layers.0.weight", "model.diffusion_model.output_blocks.11.0.out_layers.0.bias", "model.diffusion_model.output_blocks.11.0.out_layers.3.weight", "model.diffusion_model.output_blocks.11.0.out_layers.3.bias", "model.diffusion_model.output_blocks.11.0.skip_connection.weight", "model.diffusion_model.output_blocks.11.0.skip_connection.bias", "model.diffusion_model.middle_block.1.norm.weight", "model.diffusion_model.middle_block.1.norm.bias", "model.diffusion_model.middle_block.1.proj_in.weight", "model.diffusion_model.middle_block.1.proj_in.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_k.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_v.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_out.0.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.0.proj.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.ff.net.2.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_q.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_k.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_v.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.attn2.to_out.0.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.norm1.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.norm1.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.norm2.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.norm2.bias", "model.diffusion_model.middle_block.1.transformer_blocks.0.norm3.weight", "model.diffusion_model.middle_block.1.transformer_blocks.0.norm3.bias", "model.diffusion_model.middle_block.1.proj_out.weight", "model.diffusion_model.middle_block.1.proj_out.bias", "model.diffusion_model.middle_block.0.in_layers.0.weight", "model.diffusion_model.middle_block.0.in_layers.0.bias", "model.diffusion_model.middle_block.0.in_layers.2.weight", "model.diffusion_model.middle_block.0.in_layers.2.bias", "model.diffusion_model.middle_block.0.emb_layers.1.weight", "model.diffusion_model.middle_block.0.emb_layers.1.bias", "model.diffusion_model.middle_block.0.out_layers.0.weight", "model.diffusion_model.middle_block.0.out_layers.0.bias", "model.diffusion_model.middle_block.0.out_layers.3.weight", "model.diffusion_model.middle_block.0.out_layers.3.bias", "model.diffusion_model.middle_block.2.in_layers.0.weight", "model.diffusion_model.middle_block.2.in_layers.0.bias", "model.diffusion_model.middle_block.2.in_layers.2.weight", "model.diffusion_model.middle_block.2.in_layers.2.bias", "model.diffusion_model.middle_block.2.emb_layers.1.weight", "model.diffusion_model.middle_block.2.emb_layers.1.bias", "model.diffusion_model.middle_block.2.out_layers.0.weight", "model.diffusion_model.middle_block.2.out_layers.0.bias", "model.diffusion_model.middle_block.2.out_layers.3.weight", "model.diffusion_model.middle_block.2.out_layers.3.bias", "model.diffusion_model.out.0.weight", "model.diffusion_model.out.0.bias", "model.diffusion_model.out.2.weight", "model.diffusion_model.out.2.bias", "first_stage_model.encoder.conv_in.weight", "first_stage_model.encoder.conv_in.bias", "first_stage_model.encoder.down.0.block.0.norm1.weight", "first_stage_model.encoder.down.0.block.0.norm1.bias", "first_stage_model.encoder.down.0.block.0.conv1.weight", "first_stage_model.encoder.down.0.block.0.conv1.bias", "first_stage_model.encoder.down.0.block.0.norm2.weight", "first_stage_model.encoder.down.0.block.0.norm2.bias", "first_stage_model.encoder.down.0.block.0.conv2.weight", "first_stage_model.encoder.down.0.block.0.conv2.bias", "first_stage_model.encoder.down.0.block.1.norm1.weight", "first_stage_model.encoder.down.0.block.1.norm1.bias", "first_stage_model.encoder.down.0.block.1.conv1.weight", "first_stage_model.encoder.down.0.block.1.conv1.bias", "first_stage_model.encoder.down.0.block.1.norm2.weight", "first_stage_model.encoder.down.0.block.1.norm2.bias", "first_stage_model.encoder.down.0.block.1.conv2.weight", "first_stage_model.encoder.down.0.block.1.conv2.bias", "first_stage_model.encoder.down.0.downsample.conv.weight", "first_stage_model.encoder.down.0.downsample.conv.bias", "first_stage_model.encoder.down.1.block.0.norm1.weight", "first_stage_model.encoder.down.1.block.0.norm1.bias", "first_stage_model.encoder.down.1.block.0.conv1.weight", "first_stage_model.encoder.down.1.block.0.conv1.bias", "first_stage_model.encoder.down.1.block.0.norm2.weight", "first_stage_model.encoder.down.1.block.0.norm2.bias", "first_stage_model.encoder.down.1.block.0.conv2.weight", "first_stage_model.encoder.down.1.block.0.conv2.bias", "first_stage_model.encoder.down.1.block.0.nin_shortcut.weight", "first_stage_model.encoder.down.1.block.0.nin_shortcut.bias", "first_stage_model.encoder.down.1.block.1.norm1.weight", "first_stage_model.encoder.down.1.block.1.norm1.bias", "first_stage_model.encoder.down.1.block.1.conv1.weight", "first_stage_model.encoder.down.1.block.1.conv1.bias", "first_stage_model.encoder.down.1.block.1.norm2.weight", "first_stage_model.encoder.down.1.block.1.norm2.bias", "first_stage_model.encoder.down.1.block.1.conv2.weight", "first_stage_model.encoder.down.1.block.1.conv2.bias", "first_stage_model.encoder.down.1.downsample.conv.weight", "first_stage_model.encoder.down.1.downsample.conv.bias", "first_stage_model.encoder.down.2.block.0.norm1.weight", "first_stage_model.encoder.down.2.block.0.norm1.bias", "first_stage_model.encoder.down.2.block.0.conv1.weight", "first_stage_model.encoder.down.2.block.0.conv1.bias", "first_stage_model.encoder.down.2.block.0.norm2.weight", "first_stage_model.encoder.down.2.block.0.norm2.bias", "first_stage_model.encoder.down.2.block.0.conv2.weight", "first_stage_model.encoder.down.2.block.0.conv2.bias", "first_stage_model.encoder.down.2.block.0.nin_shortcut.weight", "first_stage_model.encoder.down.2.block.0.nin_shortcut.bias", "first_stage_model.encoder.down.2.block.1.norm1.weight", "first_stage_model.encoder.down.2.block.1.norm1.bias", "first_stage_model.encoder.down.2.block.1.conv1.weight", "first_stage_model.encoder.down.2.block.1.conv1.bias", "first_stage_model.encoder.down.2.block.1.norm2.weight", "first_stage_model.encoder.down.2.block.1.norm2.bias", "first_stage_model.encoder.down.2.block.1.conv2.weight", "first_stage_model.encoder.down.2.block.1.conv2.bias", "first_stage_model.encoder.down.2.downsample.conv.weight", "first_stage_model.encoder.down.2.downsample.conv.bias", "first_stage_model.encoder.down.3.block.0.norm1.weight", "first_stage_model.encoder.down.3.block.0.norm1.bias", "first_stage_model.encoder.down.3.block.0.conv1.weight", "first_stage_model.encoder.down.3.block.0.conv1.bias", "first_stage_model.encoder.down.3.block.0.norm2.weight", "first_stage_model.encoder.down.3.block.0.norm2.bias", "first_stage_model.encoder.down.3.block.0.conv2.weight", "first_stage_model.encoder.down.3.block.0.conv2.bias", "first_stage_model.encoder.down.3.block.1.norm1.weight", "first_stage_model.encoder.down.3.block.1.norm1.bias", "first_stage_model.encoder.down.3.block.1.conv1.weight", "first_stage_model.encoder.down.3.block.1.conv1.bias", "first_stage_model.encoder.down.3.block.1.norm2.weight", "first_stage_model.encoder.down.3.block.1.norm2.bias", "first_stage_model.encoder.down.3.block.1.conv2.weight", "first_stage_model.encoder.down.3.block.1.conv2.bias", "first_stage_model.encoder.mid.attn_1.norm.weight", "first_stage_model.encoder.mid.attn_1.norm.bias", "first_stage_model.encoder.mid.attn_1.q.weight", "first_stage_model.encoder.mid.attn_1.q.bias", "first_stage_model.encoder.mid.attn_1.k.weight", "first_stage_model.encoder.mid.attn_1.k.bias", "first_stage_model.encoder.mid.attn_1.v.weight", "first_stage_model.encoder.mid.attn_1.v.bias", "first_stage_model.encoder.mid.attn_1.proj_out.weight", "first_stage_model.encoder.mid.attn_1.proj_out.bias", "first_stage_model.encoder.mid.block_1.norm1.weight", "first_stage_model.encoder.mid.block_1.norm1.bias", "first_stage_model.encoder.mid.block_1.conv1.weight", "first_stage_model.encoder.mid.block_1.conv1.bias", "first_stage_model.encoder.mid.block_1.norm2.weight", "first_stage_model.encoder.mid.block_1.norm2.bias", "first_stage_model.encoder.mid.block_1.conv2.weight", "first_stage_model.encoder.mid.block_1.conv2.bias", "first_stage_model.encoder.mid.block_2.norm1.weight", "first_stage_model.encoder.mid.block_2.norm1.bias", "first_stage_model.encoder.mid.block_2.conv1.weight", "first_stage_model.encoder.mid.block_2.conv1.bias", "first_stage_model.encoder.mid.block_2.norm2.weight", "first_stage_model.encoder.mid.block_2.norm2.bias", "first_stage_model.encoder.mid.block_2.conv2.weight", "first_stage_model.encoder.mid.block_2.conv2.bias", "first_stage_model.encoder.norm_out.weight", "first_stage_model.encoder.norm_out.bias", "first_stage_model.encoder.conv_out.weight", "first_stage_model.encoder.conv_out.bias", "first_stage_model.decoder.conv_in.weight", "first_stage_model.decoder.conv_in.bias", "first_stage_model.decoder.up.3.block.0.norm1.weight", "first_stage_model.decoder.up.3.block.0.norm1.bias", "first_stage_model.decoder.up.3.block.0.conv1.weight", "first_stage_model.decoder.up.3.block.0.conv1.bias", "first_stage_model.decoder.up.3.block.0.norm2.weight", "first_stage_model.decoder.up.3.block.0.norm2.bias", "first_stage_model.decoder.up.3.block.0.conv2.weight", "first_stage_model.decoder.up.3.block.0.conv2.bias", "first_stage_model.decoder.up.3.block.1.norm1.weight", "first_stage_model.decoder.up.3.block.1.norm1.bias", "first_stage_model.decoder.up.3.block.1.conv1.weight", "first_stage_model.decoder.up.3.block.1.conv1.bias", "first_stage_model.decoder.up.3.block.1.norm2.weight", "first_stage_model.decoder.up.3.block.1.norm2.bias", "first_stage_model.decoder.up.3.block.1.conv2.weight", "first_stage_model.decoder.up.3.block.1.conv2.bias", "first_stage_model.decoder.up.3.block.2.norm1.weight", "first_stage_model.decoder.up.3.block.2.norm1.bias", "first_stage_model.decoder.up.3.block.2.conv1.weight", "first_stage_model.decoder.up.3.block.2.conv1.bias", "first_stage_model.decoder.up.3.block.2.norm2.weight", "first_stage_model.decoder.up.3.block.2.norm2.bias", "first_stage_model.decoder.up.3.block.2.conv2.weight", "first_stage_model.decoder.up.3.block.2.conv2.bias", "first_stage_model.decoder.up.3.upsample.conv.weight", "first_stage_model.decoder.up.3.upsample.conv.bias", "first_stage_model.decoder.up.2.block.0.norm1.weight", "first_stage_model.decoder.up.2.block.0.norm1.bias", "first_stage_model.decoder.up.2.block.0.conv1.weight", "first_stage_model.decoder.up.2.block.0.conv1.bias", "first_stage_model.decoder.up.2.block.0.norm2.weight", "first_stage_model.decoder.up.2.block.0.norm2.bias", "first_stage_model.decoder.up.2.block.0.conv2.weight", "first_stage_model.decoder.up.2.block.0.conv2.bias", "first_stage_model.decoder.up.2.block.1.norm1.weight", "first_stage_model.decoder.up.2.block.1.norm1.bias", "first_stage_model.decoder.up.2.block.1.conv1.weight", "first_stage_model.decoder.up.2.block.1.conv1.bias", "first_stage_model.decoder.up.2.block.1.norm2.weight", "first_stage_model.decoder.up.2.block.1.norm2.bias", "first_stage_model.decoder.up.2.block.1.conv2.weight", "first_stage_model.decoder.up.2.block.1.conv2.bias", "first_stage_model.decoder.up.2.block.2.norm1.weight", "first_stage_model.decoder.up.2.block.2.norm1.bias", "first_stage_model.decoder.up.2.block.2.conv1.weight", "first_stage_model.decoder.up.2.block.2.conv1.bias", "first_stage_model.decoder.up.2.block.2.norm2.weight", "first_stage_model.decoder.up.2.block.2.norm2.bias", "first_stage_model.decoder.up.2.block.2.conv2.weight", "first_stage_model.decoder.up.2.block.2.conv2.bias", "first_stage_model.decoder.up.2.upsample.conv.weight", "first_stage_model.decoder.up.2.upsample.conv.bias", "first_stage_model.decoder.up.1.block.0.norm1.weight", "first_stage_model.decoder.up.1.block.0.norm1.bias", "first_stage_model.decoder.up.1.block.0.conv1.weight", "first_stage_model.decoder.up.1.block.0.conv1.bias", "first_stage_model.decoder.up.1.block.0.norm2.weight", "first_stage_model.decoder.up.1.block.0.norm2.bias", "first_stage_model.decoder.up.1.block.0.conv2.weight", "first_stage_model.decoder.up.1.block.0.conv2.bias", "first_stage_model.decoder.up.1.block.0.nin_shortcut.weight", "first_stage_model.decoder.up.1.block.0.nin_shortcut.bias", "first_stage_model.decoder.up.1.block.1.norm1.weight", "first_stage_model.decoder.up.1.block.1.norm1.bias", "first_stage_model.decoder.up.1.block.1.conv1.weight", "first_stage_model.decoder.up.1.block.1.conv1.bias", "first_stage_model.decoder.up.1.block.1.norm2.weight", "first_stage_model.decoder.up.1.block.1.norm2.bias", "first_stage_model.decoder.up.1.block.1.conv2.weight", "first_stage_model.decoder.up.1.block.1.conv2.bias", "first_stage_model.decoder.up.1.block.2.norm1.weight", "first_stage_model.decoder.up.1.block.2.norm1.bias", "first_stage_model.decoder.up.1.block.2.conv1.weight", "first_stage_model.decoder.up.1.block.2.conv1.bias", "first_stage_model.decoder.up.1.block.2.norm2.weight", "first_stage_model.decoder.up.1.block.2.norm2.bias", "first_stage_model.decoder.up.1.block.2.conv2.weight", "first_stage_model.decoder.up.1.block.2.conv2.bias", "first_stage_model.decoder.up.1.upsample.conv.weight", "first_stage_model.decoder.up.1.upsample.conv.bias", "first_stage_model.decoder.up.0.block.0.norm1.weight", "first_stage_model.decoder.up.0.block.0.norm1.bias", "first_stage_model.decoder.up.0.block.0.conv1.weight", "first_stage_model.decoder.up.0.block.0.conv1.bias", "first_stage_model.decoder.up.0.block.0.norm2.weight", "first_stage_model.decoder.up.0.block.0.norm2.bias", "first_stage_model.decoder.up.0.block.0.conv2.weight", "first_stage_model.decoder.up.0.block.0.conv2.bias", "first_stage_model.decoder.up.0.block.0.nin_shortcut.weight", "first_stage_model.decoder.up.0.block.0.nin_shortcut.bias", "first_stage_model.decoder.up.0.block.1.norm1.weight", "first_stage_model.decoder.up.0.block.1.norm1.bias", "first_stage_model.decoder.up.0.block.1.conv1.weight", "first_stage_model.decoder.up.0.block.1.conv1.bias", "first_stage_model.decoder.up.0.block.1.norm2.weight", "first_stage_model.decoder.up.0.block.1.norm2.bias", "first_stage_model.decoder.up.0.block.1.conv2.weight", "first_stage_model.decoder.up.0.block.1.conv2.bias", "first_stage_model.decoder.up.0.block.2.norm1.weight", "first_stage_model.decoder.up.0.block.2.norm1.bias", "first_stage_model.decoder.up.0.block.2.conv1.weight", "first_stage_model.decoder.up.0.block.2.conv1.bias", "first_stage_model.decoder.up.0.block.2.norm2.weight", "first_stage_model.decoder.up.0.block.2.norm2.bias", "first_stage_model.decoder.up.0.block.2.conv2.weight", "first_stage_model.decoder.up.0.block.2.conv2.bias", "first_stage_model.decoder.mid.attn_1.norm.weight", "first_stage_model.decoder.mid.attn_1.norm.bias", "first_stage_model.decoder.mid.attn_1.q.weight", "first_stage_model.decoder.mid.attn_1.q.bias", "first_stage_model.decoder.mid.attn_1.k.weight", "first_stage_model.decoder.mid.attn_1.k.bias", "first_stage_model.decoder.mid.attn_1.v.weight", "first_stage_model.decoder.mid.attn_1.v.bias", "first_stage_model.decoder.mid.attn_1.proj_out.weight", "first_stage_model.decoder.mid.attn_1.proj_out.bias", "first_stage_model.decoder.mid.block_1.norm1.weight", "first_stage_model.decoder.mid.block_1.norm1.bias", "first_stage_model.decoder.mid.block_1.conv1.weight", "first_stage_model.decoder.mid.block_1.conv1.bias", "first_stage_model.decoder.mid.block_1.norm2.weight", "first_stage_model.decoder.mid.block_1.norm2.bias", "first_stage_model.decoder.mid.block_1.conv2.weight", "first_stage_model.decoder.mid.block_1.conv2.bias", "first_stage_model.decoder.mid.block_2.norm1.weight", "first_stage_model.decoder.mid.block_2.norm1.bias", "first_stage_model.decoder.mid.block_2.conv1.weight", "first_stage_model.decoder.mid.block_2.conv1.bias", "first_stage_model.decoder.mid.block_2.norm2.weight", "first_stage_model.decoder.mid.block_2.norm2.bias", "first_stage_model.decoder.mid.block_2.conv2.weight", "first_stage_model.decoder.mid.block_2.conv2.bias", "first_stage_model.decoder.norm_out.weight", "first_stage_model.decoder.norm_out.bias", "first_stage_model.decoder.conv_out.weight", "first_stage_model.decoder.conv_out.bias", "first_stage_model.quant_conv.weight", "first_stage_model.quant_conv.bias", "first_stage_model.post_quant_conv.weight", "first_stage_model.post_quant_conv.bias", "cond_stage_model.transformer.text_model.embeddings.position_ids", "cond_stage_model.transformer.text_model.embeddings.token_embedding.weight", "cond_stage_model.transformer.text_model.embeddings.position_embedding.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.0.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.1.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.2.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.3.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.4.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.5.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.6.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.7.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.8.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.9.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.10.layer_norm2.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.k_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.k_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.v_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.v_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.q_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.q_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.out_proj.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.self_attn.out_proj.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm1.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm1.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc1.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc1.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc2.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.mlp.fc2.bias", "cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm2.weight", "cond_stage_model.transformer.text_model.encoder.layers.11.layer_norm2.bias", "cond_stage_model.transformer.text_model.final_layer_norm.weight", "cond_stage_model.transformer.text_model.final_layer_norm.bias", "model_ema.diffusion_modeltime_embed0weight", "model_ema.diffusion_modeltime_embed0bias", "model_ema.diffusion_modeltime_embed2weight", "model_ema.diffusion_modeltime_embed2bias", "model_ema.diffusion_modelinput_blocks00weight", "model_ema.diffusion_modelinput_blocks00bias", "model_ema.diffusion_modelinput_blocks10in_layers0weight", "model_ema.diffusion_modelinput_blocks10in_layers0bias", "model_ema.diffusion_modelinput_blocks10in_layers2weight", "model_ema.diffusion_modelinput_blocks10in_layers2bias", "model_ema.diffusion_modelinput_blocks10emb_layers1weight", "model_ema.diffusion_modelinput_blocks10emb_layers1bias", "model_ema.diffusion_modelinput_blocks10out_layers0weight", "model_ema.diffusion_modelinput_blocks10out_layers0bias", "model_ema.diffusion_modelinput_blocks10out_layers3weight", "model_ema.diffusion_modelinput_blocks10out_layers3bias", "model_ema.diffusion_modelinput_blocks11normweight", "model_ema.diffusion_modelinput_blocks11normbias", "model_ema.diffusion_modelinput_blocks11proj_inweight", "model_ema.diffusion_modelinput_blocks11proj_inbias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0ffnet2weight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0ffnet2bias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0norm1weight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0norm1bias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0norm2weight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0norm2bias", "model_ema.diffusion_modelinput_blocks11transformer_blocks0norm3weight", "model_ema.diffusion_modelinput_blocks11transformer_blocks0norm3bias", "model_ema.diffusion_modelinput_blocks11proj_outweight", "model_ema.diffusion_modelinput_blocks11proj_outbias", "model_ema.diffusion_modelinput_blocks20in_layers0weight", "model_ema.diffusion_modelinput_blocks20in_layers0bias", "model_ema.diffusion_modelinput_blocks20in_layers2weight", "model_ema.diffusion_modelinput_blocks20in_layers2bias", "model_ema.diffusion_modelinput_blocks20emb_layers1weight", "model_ema.diffusion_modelinput_blocks20emb_layers1bias", "model_ema.diffusion_modelinput_blocks20out_layers0weight", "model_ema.diffusion_modelinput_blocks20out_layers0bias", "model_ema.diffusion_modelinput_blocks20out_layers3weight", "model_ema.diffusion_modelinput_blocks20out_layers3bias", "model_ema.diffusion_modelinput_blocks21normweight", "model_ema.diffusion_modelinput_blocks21normbias", "model_ema.diffusion_modelinput_blocks21proj_inweight", "model_ema.diffusion_modelinput_blocks21proj_inbias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0ffnet2weight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0ffnet2bias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0norm1weight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0norm1bias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0norm2weight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0norm2bias", "model_ema.diffusion_modelinput_blocks21transformer_blocks0norm3weight", "model_ema.diffusion_modelinput_blocks21transformer_blocks0norm3bias", "model_ema.diffusion_modelinput_blocks21proj_outweight", "model_ema.diffusion_modelinput_blocks21proj_outbias", "model_ema.diffusion_modelinput_blocks30opweight", "model_ema.diffusion_modelinput_blocks30opbias", "model_ema.diffusion_modelinput_blocks40in_layers0weight", "model_ema.diffusion_modelinput_blocks40in_layers0bias", "model_ema.diffusion_modelinput_blocks40in_layers2weight", "model_ema.diffusion_modelinput_blocks40in_layers2bias", "model_ema.diffusion_modelinput_blocks40emb_layers1weight", "model_ema.diffusion_modelinput_blocks40emb_layers1bias", "model_ema.diffusion_modelinput_blocks40out_layers0weight", "model_ema.diffusion_modelinput_blocks40out_layers0bias", "model_ema.diffusion_modelinput_blocks40out_layers3weight", "model_ema.diffusion_modelinput_blocks40out_layers3bias", "model_ema.diffusion_modelinput_blocks40skip_connectionweight", "model_ema.diffusion_modelinput_blocks40skip_connectionbias", "model_ema.diffusion_modelinput_blocks41normweight", "model_ema.diffusion_modelinput_blocks41normbias", "model_ema.diffusion_modelinput_blocks41proj_inweight", "model_ema.diffusion_modelinput_blocks41proj_inbias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0ffnet2weight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0ffnet2bias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0norm1weight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0norm1bias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0norm2weight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0norm2bias", "model_ema.diffusion_modelinput_blocks41transformer_blocks0norm3weight", "model_ema.diffusion_modelinput_blocks41transformer_blocks0norm3bias", "model_ema.diffusion_modelinput_blocks41proj_outweight", "model_ema.diffusion_modelinput_blocks41proj_outbias", "model_ema.diffusion_modelinput_blocks50in_layers0weight", "model_ema.diffusion_modelinput_blocks50in_layers0bias", "model_ema.diffusion_modelinput_blocks50in_layers2weight", "model_ema.diffusion_modelinput_blocks50in_layers2bias", "model_ema.diffusion_modelinput_blocks50emb_layers1weight", "model_ema.diffusion_modelinput_blocks50emb_layers1bias", "model_ema.diffusion_modelinput_blocks50out_layers0weight", "model_ema.diffusion_modelinput_blocks50out_layers0bias", "model_ema.diffusion_modelinput_blocks50out_layers3weight", "model_ema.diffusion_modelinput_blocks50out_layers3bias", "model_ema.diffusion_modelinput_blocks51normweight", "model_ema.diffusion_modelinput_blocks51normbias", "model_ema.diffusion_modelinput_blocks51proj_inweight", "model_ema.diffusion_modelinput_blocks51proj_inbias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0ffnet2weight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0ffnet2bias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0norm1weight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0norm1bias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0norm2weight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0norm2bias", "model_ema.diffusion_modelinput_blocks51transformer_blocks0norm3weight", "model_ema.diffusion_modelinput_blocks51transformer_blocks0norm3bias", "model_ema.diffusion_modelinput_blocks51proj_outweight", "model_ema.diffusion_modelinput_blocks51proj_outbias", "model_ema.diffusion_modelinput_blocks60opweight", "model_ema.diffusion_modelinput_blocks60opbias", "model_ema.diffusion_modelinput_blocks70in_layers0weight", "model_ema.diffusion_modelinput_blocks70in_layers0bias", "model_ema.diffusion_modelinput_blocks70in_layers2weight", "model_ema.diffusion_modelinput_blocks70in_layers2bias", "model_ema.diffusion_modelinput_blocks70emb_layers1weight", "model_ema.diffusion_modelinput_blocks70emb_layers1bias", "model_ema.diffusion_modelinput_blocks70out_layers0weight", "model_ema.diffusion_modelinput_blocks70out_layers0bias", "model_ema.diffusion_modelinput_blocks70out_layers3weight", "model_ema.diffusion_modelinput_blocks70out_layers3bias", "model_ema.diffusion_modelinput_blocks70skip_connectionweight", "model_ema.diffusion_modelinput_blocks70skip_connectionbias", "model_ema.diffusion_modelinput_blocks71normweight", "model_ema.diffusion_modelinput_blocks71normbias", "model_ema.diffusion_modelinput_blocks71proj_inweight", "model_ema.diffusion_modelinput_blocks71proj_inbias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0ffnet2weight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0ffnet2bias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0norm1weight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0norm1bias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0norm2weight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0norm2bias", "model_ema.diffusion_modelinput_blocks71transformer_blocks0norm3weight", "model_ema.diffusion_modelinput_blocks71transformer_blocks0norm3bias", "model_ema.diffusion_modelinput_blocks71proj_outweight", "model_ema.diffusion_modelinput_blocks71proj_outbias", "model_ema.diffusion_modelinput_blocks80in_layers0weight", "model_ema.diffusion_modelinput_blocks80in_layers0bias", "model_ema.diffusion_modelinput_blocks80in_layers2weight", "model_ema.diffusion_modelinput_blocks80in_layers2bias", "model_ema.diffusion_modelinput_blocks80emb_layers1weight", "model_ema.diffusion_modelinput_blocks80emb_layers1bias", "model_ema.diffusion_modelinput_blocks80out_layers0weight", "model_ema.diffusion_modelinput_blocks80out_layers0bias", "model_ema.diffusion_modelinput_blocks80out_layers3weight", "model_ema.diffusion_modelinput_blocks80out_layers3bias", "model_ema.diffusion_modelinput_blocks81normweight", "model_ema.diffusion_modelinput_blocks81normbias", "model_ema.diffusion_modelinput_blocks81proj_inweight", "model_ema.diffusion_modelinput_blocks81proj_inbias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0ffnet2weight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0ffnet2bias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0norm1weight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0norm1bias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0norm2weight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0norm2bias", "model_ema.diffusion_modelinput_blocks81transformer_blocks0norm3weight", "model_ema.diffusion_modelinput_blocks81transformer_blocks0norm3bias", "model_ema.diffusion_modelinput_blocks81proj_outweight", "model_ema.diffusion_modelinput_blocks81proj_outbias", "model_ema.diffusion_modelinput_blocks90opweight", "model_ema.diffusion_modelinput_blocks90opbias", "model_ema.diffusion_modelinput_blocks100in_layers0weight", "model_ema.diffusion_modelinput_blocks100in_layers0bias", "model_ema.diffusion_modelinput_blocks100in_layers2weight", "model_ema.diffusion_modelinput_blocks100in_layers2bias", "model_ema.diffusion_modelinput_blocks100emb_layers1weight", "model_ema.diffusion_modelinput_blocks100emb_layers1bias", "model_ema.diffusion_modelinput_blocks100out_layers0weight", "model_ema.diffusion_modelinput_blocks100out_layers0bias", "model_ema.diffusion_modelinput_blocks100out_layers3weight", "model_ema.diffusion_modelinput_blocks100out_layers3bias", "model_ema.diffusion_modelinput_blocks110in_layers0weight", "model_ema.diffusion_modelinput_blocks110in_layers0bias", "model_ema.diffusion_modelinput_blocks110in_layers2weight", "model_ema.diffusion_modelinput_blocks110in_layers2bias", "model_ema.diffusion_modelinput_blocks110emb_layers1weight", "model_ema.diffusion_modelinput_blocks110emb_layers1bias", "model_ema.diffusion_modelinput_blocks110out_layers0weight", "model_ema.diffusion_modelinput_blocks110out_layers0bias", "model_ema.diffusion_modelinput_blocks110out_layers3weight", "model_ema.diffusion_modelinput_blocks110out_layers3bias", "model_ema.diffusion_modelmiddle_block0in_layers0weight", "model_ema.diffusion_modelmiddle_block0in_layers0bias", "model_ema.diffusion_modelmiddle_block0in_layers2weight", "model_ema.diffusion_modelmiddle_block0in_layers2bias", "model_ema.diffusion_modelmiddle_block0emb_layers1weight", "model_ema.diffusion_modelmiddle_block0emb_layers1bias", "model_ema.diffusion_modelmiddle_block0out_layers0weight", "model_ema.diffusion_modelmiddle_block0out_layers0bias", "model_ema.diffusion_modelmiddle_block0out_layers3weight", "model_ema.diffusion_modelmiddle_block0out_layers3bias", "model_ema.diffusion_modelmiddle_block1normweight", "model_ema.diffusion_modelmiddle_block1normbias", "model_ema.diffusion_modelmiddle_block1proj_inweight", "model_ema.diffusion_modelmiddle_block1proj_inbias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn1to_qweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn1to_kweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn1to_vweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0ffnet0projweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0ffnet0projbias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0ffnet2weight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0ffnet2bias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn2to_qweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn2to_kweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn2to_vweight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0norm1weight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0norm1bias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0norm2weight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0norm2bias", "model_ema.diffusion_modelmiddle_block1transformer_blocks0norm3weight", "model_ema.diffusion_modelmiddle_block1transformer_blocks0norm3bias", "model_ema.diffusion_modelmiddle_block1proj_outweight", "model_ema.diffusion_modelmiddle_block1proj_outbias", "model_ema.diffusion_modelmiddle_block2in_layers0weight", "model_ema.diffusion_modelmiddle_block2in_layers0bias", "model_ema.diffusion_modelmiddle_block2in_layers2weight", "model_ema.diffusion_modelmiddle_block2in_layers2bias", "model_ema.diffusion_modelmiddle_block2emb_layers1weight", "model_ema.diffusion_modelmiddle_block2emb_layers1bias", "model_ema.diffusion_modelmiddle_block2out_layers0weight", "model_ema.diffusion_modelmiddle_block2out_layers0bias", "model_ema.diffusion_modelmiddle_block2out_layers3weight", "model_ema.diffusion_modelmiddle_block2out_layers3bias", "model_ema.diffusion_modeloutput_blocks00in_layers0weight", "model_ema.diffusion_modeloutput_blocks00in_layers0bias", "model_ema.diffusion_modeloutput_blocks00in_layers2weight", "model_ema.diffusion_modeloutput_blocks00in_layers2bias", "model_ema.diffusion_modeloutput_blocks00emb_layers1weight", "model_ema.diffusion_modeloutput_blocks00emb_layers1bias", "model_ema.diffusion_modeloutput_blocks00out_layers0weight", "model_ema.diffusion_modeloutput_blocks00out_layers0bias", "model_ema.diffusion_modeloutput_blocks00out_layers3weight", "model_ema.diffusion_modeloutput_blocks00out_layers3bias", "model_ema.diffusion_modeloutput_blocks00skip_connectionweight", "model_ema.diffusion_modeloutput_blocks00skip_connectionbias", "model_ema.diffusion_modeloutput_blocks10in_layers0weight", "model_ema.diffusion_modeloutput_blocks10in_layers0bias", "model_ema.diffusion_modeloutput_blocks10in_layers2weight", "model_ema.diffusion_modeloutput_blocks10in_layers2bias", "model_ema.diffusion_modeloutput_blocks10emb_layers1weight", "model_ema.diffusion_modeloutput_blocks10emb_layers1bias", "model_ema.diffusion_modeloutput_blocks10out_layers0weight", "model_ema.diffusion_modeloutput_blocks10out_layers0bias", "model_ema.diffusion_modeloutput_blocks10out_layers3weight", "model_ema.diffusion_modeloutput_blocks10out_layers3bias", "model_ema.diffusion_modeloutput_blocks10skip_connectionweight", "model_ema.diffusion_modeloutput_blocks10skip_connectionbias", "model_ema.diffusion_modeloutput_blocks20in_layers0weight", "model_ema.diffusion_modeloutput_blocks20in_layers0bias", "model_ema.diffusion_modeloutput_blocks20in_layers2weight", "model_ema.diffusion_modeloutput_blocks20in_layers2bias", "model_ema.diffusion_modeloutput_blocks20emb_layers1weight", "model_ema.diffusion_modeloutput_blocks20emb_layers1bias", "model_ema.diffusion_modeloutput_blocks20out_layers0weight", "model_ema.diffusion_modeloutput_blocks20out_layers0bias", "model_ema.diffusion_modeloutput_blocks20out_layers3weight", "model_ema.diffusion_modeloutput_blocks20out_layers3bias", "model_ema.diffusion_modeloutput_blocks20skip_connectionweight", "model_ema.diffusion_modeloutput_blocks20skip_connectionbias", "model_ema.diffusion_modeloutput_blocks21convweight", "model_ema.diffusion_modeloutput_blocks21convbias", "model_ema.diffusion_modeloutput_blocks30in_layers0weight", "model_ema.diffusion_modeloutput_blocks30in_layers0bias", "model_ema.diffusion_modeloutput_blocks30in_layers2weight", "model_ema.diffusion_modeloutput_blocks30in_layers2bias", "model_ema.diffusion_modeloutput_blocks30emb_layers1weight", "model_ema.diffusion_modeloutput_blocks30emb_layers1bias", "model_ema.diffusion_modeloutput_blocks30out_layers0weight", "model_ema.diffusion_modeloutput_blocks30out_layers0bias", "model_ema.diffusion_modeloutput_blocks30out_layers3weight", "model_ema.diffusion_modeloutput_blocks30out_layers3bias", "model_ema.diffusion_modeloutput_blocks30skip_connectionweight", "model_ema.diffusion_modeloutput_blocks30skip_connectionbias", "model_ema.diffusion_modeloutput_blocks31normweight", "model_ema.diffusion_modeloutput_blocks31normbias", "model_ema.diffusion_modeloutput_blocks31proj_inweight", "model_ema.diffusion_modeloutput_blocks31proj_inbias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks31transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks31proj_outweight", "model_ema.diffusion_modeloutput_blocks31proj_outbias", "model_ema.diffusion_modeloutput_blocks40in_layers0weight", "model_ema.diffusion_modeloutput_blocks40in_layers0bias", "model_ema.diffusion_modeloutput_blocks40in_layers2weight", "model_ema.diffusion_modeloutput_blocks40in_layers2bias", "model_ema.diffusion_modeloutput_blocks40emb_layers1weight", "model_ema.diffusion_modeloutput_blocks40emb_layers1bias", "model_ema.diffusion_modeloutput_blocks40out_layers0weight", "model_ema.diffusion_modeloutput_blocks40out_layers0bias", "model_ema.diffusion_modeloutput_blocks40out_layers3weight", "model_ema.diffusion_modeloutput_blocks40out_layers3bias", "model_ema.diffusion_modeloutput_blocks40skip_connectionweight", "model_ema.diffusion_modeloutput_blocks40skip_connectionbias", "model_ema.diffusion_modeloutput_blocks41normweight", "model_ema.diffusion_modeloutput_blocks41normbias", "model_ema.diffusion_modeloutput_blocks41proj_inweight", "model_ema.diffusion_modeloutput_blocks41proj_inbias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks41transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks41proj_outweight", "model_ema.diffusion_modeloutput_blocks41proj_outbias", "model_ema.diffusion_modeloutput_blocks50in_layers0weight", "model_ema.diffusion_modeloutput_blocks50in_layers0bias", "model_ema.diffusion_modeloutput_blocks50in_layers2weight", "model_ema.diffusion_modeloutput_blocks50in_layers2bias", "model_ema.diffusion_modeloutput_blocks50emb_layers1weight", "model_ema.diffusion_modeloutput_blocks50emb_layers1bias", "model_ema.diffusion_modeloutput_blocks50out_layers0weight", "model_ema.diffusion_modeloutput_blocks50out_layers0bias", "model_ema.diffusion_modeloutput_blocks50out_layers3weight", "model_ema.diffusion_modeloutput_blocks50out_layers3bias", "model_ema.diffusion_modeloutput_blocks50skip_connectionweight", "model_ema.diffusion_modeloutput_blocks50skip_connectionbias", "model_ema.diffusion_modeloutput_blocks51normweight", "model_ema.diffusion_modeloutput_blocks51normbias", "model_ema.diffusion_modeloutput_blocks51proj_inweight", "model_ema.diffusion_modeloutput_blocks51proj_inbias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks51transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks51proj_outweight", "model_ema.diffusion_modeloutput_blocks51proj_outbias", "model_ema.diffusion_modeloutput_blocks52convweight", "model_ema.diffusion_modeloutput_blocks52convbias", "model_ema.diffusion_modeloutput_blocks60in_layers0weight", "model_ema.diffusion_modeloutput_blocks60in_layers0bias", "model_ema.diffusion_modeloutput_blocks60in_layers2weight", "model_ema.diffusion_modeloutput_blocks60in_layers2bias", "model_ema.diffusion_modeloutput_blocks60emb_layers1weight", "model_ema.diffusion_modeloutput_blocks60emb_layers1bias", "model_ema.diffusion_modeloutput_blocks60out_layers0weight", "model_ema.diffusion_modeloutput_blocks60out_layers0bias", "model_ema.diffusion_modeloutput_blocks60out_layers3weight", "model_ema.diffusion_modeloutput_blocks60out_layers3bias", "model_ema.diffusion_modeloutput_blocks60skip_connectionweight", "model_ema.diffusion_modeloutput_blocks60skip_connectionbias", "model_ema.diffusion_modeloutput_blocks61normweight", "model_ema.diffusion_modeloutput_blocks61normbias", "model_ema.diffusion_modeloutput_blocks61proj_inweight", "model_ema.diffusion_modeloutput_blocks61proj_inbias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks61transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks61proj_outweight", "model_ema.diffusion_modeloutput_blocks61proj_outbias", "model_ema.diffusion_modeloutput_blocks70in_layers0weight", "model_ema.diffusion_modeloutput_blocks70in_layers0bias", "model_ema.diffusion_modeloutput_blocks70in_layers2weight", "model_ema.diffusion_modeloutput_blocks70in_layers2bias", "model_ema.diffusion_modeloutput_blocks70emb_layers1weight", "model_ema.diffusion_modeloutput_blocks70emb_layers1bias", "model_ema.diffusion_modeloutput_blocks70out_layers0weight", "model_ema.diffusion_modeloutput_blocks70out_layers0bias", "model_ema.diffusion_modeloutput_blocks70out_layers3weight", "model_ema.diffusion_modeloutput_blocks70out_layers3bias", "model_ema.diffusion_modeloutput_blocks70skip_connectionweight", "model_ema.diffusion_modeloutput_blocks70skip_connectionbias", "model_ema.diffusion_modeloutput_blocks71normweight", "model_ema.diffusion_modeloutput_blocks71normbias", "model_ema.diffusion_modeloutput_blocks71proj_inweight", "model_ema.diffusion_modeloutput_blocks71proj_inbias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks71transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks71proj_outweight", "model_ema.diffusion_modeloutput_blocks71proj_outbias", "model_ema.diffusion_modeloutput_blocks80in_layers0weight", "model_ema.diffusion_modeloutput_blocks80in_layers0bias", "model_ema.diffusion_modeloutput_blocks80in_layers2weight", "model_ema.diffusion_modeloutput_blocks80in_layers2bias", "model_ema.diffusion_modeloutput_blocks80emb_layers1weight", "model_ema.diffusion_modeloutput_blocks80emb_layers1bias", "model_ema.diffusion_modeloutput_blocks80out_layers0weight", "model_ema.diffusion_modeloutput_blocks80out_layers0bias", "model_ema.diffusion_modeloutput_blocks80out_layers3weight", "model_ema.diffusion_modeloutput_blocks80out_layers3bias", "model_ema.diffusion_modeloutput_blocks80skip_connectionweight", "model_ema.diffusion_modeloutput_blocks80skip_connectionbias", "model_ema.diffusion_modeloutput_blocks81normweight", "model_ema.diffusion_modeloutput_blocks81normbias", "model_ema.diffusion_modeloutput_blocks81proj_inweight", "model_ema.diffusion_modeloutput_blocks81proj_inbias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks81transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks81proj_outweight", "model_ema.diffusion_modeloutput_blocks81proj_outbias", "model_ema.diffusion_modeloutput_blocks82convweight", "model_ema.diffusion_modeloutput_blocks82convbias", "model_ema.diffusion_modeloutput_blocks90in_layers0weight", "model_ema.diffusion_modeloutput_blocks90in_layers0bias", "model_ema.diffusion_modeloutput_blocks90in_layers2weight", "model_ema.diffusion_modeloutput_blocks90in_layers2bias", "model_ema.diffusion_modeloutput_blocks90emb_layers1weight", "model_ema.diffusion_modeloutput_blocks90emb_layers1bias", "model_ema.diffusion_modeloutput_blocks90out_layers0weight", "model_ema.diffusion_modeloutput_blocks90out_layers0bias", "model_ema.diffusion_modeloutput_blocks90out_layers3weight", "model_ema.diffusion_modeloutput_blocks90out_layers3bias", "model_ema.diffusion_modeloutput_blocks90skip_connectionweight", "model_ema.diffusion_modeloutput_blocks90skip_connectionbias", "model_ema.diffusion_modeloutput_blocks91normweight", "model_ema.diffusion_modeloutput_blocks91normbias", "model_ema.diffusion_modeloutput_blocks91proj_inweight", "model_ema.diffusion_modeloutput_blocks91proj_inbias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks91transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks91proj_outweight", "model_ema.diffusion_modeloutput_blocks91proj_outbias", "model_ema.diffusion_modeloutput_blocks100in_layers0weight", "model_ema.diffusion_modeloutput_blocks100in_layers0bias", "model_ema.diffusion_modeloutput_blocks100in_layers2weight", "model_ema.diffusion_modeloutput_blocks100in_layers2bias", "model_ema.diffusion_modeloutput_blocks100emb_layers1weight", "model_ema.diffusion_modeloutput_blocks100emb_layers1bias", "model_ema.diffusion_modeloutput_blocks100out_layers0weight", "model_ema.diffusion_modeloutput_blocks100out_layers0bias", "model_ema.diffusion_modeloutput_blocks100out_layers3weight", "model_ema.diffusion_modeloutput_blocks100out_layers3bias", "model_ema.diffusion_modeloutput_blocks100skip_connectionweight", "model_ema.diffusion_modeloutput_blocks100skip_connectionbias", "model_ema.diffusion_modeloutput_blocks101normweight", "model_ema.diffusion_modeloutput_blocks101normbias", "model_ema.diffusion_modeloutput_blocks101proj_inweight", "model_ema.diffusion_modeloutput_blocks101proj_inbias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks101transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks101proj_outweight", "model_ema.diffusion_modeloutput_blocks101proj_outbias", "model_ema.diffusion_modeloutput_blocks110in_layers0weight", "model_ema.diffusion_modeloutput_blocks110in_layers0bias", "model_ema.diffusion_modeloutput_blocks110in_layers2weight", "model_ema.diffusion_modeloutput_blocks110in_layers2bias", "model_ema.diffusion_modeloutput_blocks110emb_layers1weight", "model_ema.diffusion_modeloutput_blocks110emb_layers1bias", "model_ema.diffusion_modeloutput_blocks110out_layers0weight", "model_ema.diffusion_modeloutput_blocks110out_layers0bias", "model_ema.diffusion_modeloutput_blocks110out_layers3weight", "model_ema.diffusion_modeloutput_blocks110out_layers3bias", "model_ema.diffusion_modeloutput_blocks110skip_connectionweight", "model_ema.diffusion_modeloutput_blocks110skip_connectionbias", "model_ema.diffusion_modeloutput_blocks111normweight", "model_ema.diffusion_modeloutput_blocks111normbias", "model_ema.diffusion_modeloutput_blocks111proj_inweight", "model_ema.diffusion_modeloutput_blocks111proj_inbias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn1to_qweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn1to_kweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn1to_vweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn1to_out0weight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn1to_out0bias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0ffnet0projweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0ffnet0projbias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0ffnet2weight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0ffnet2bias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn2to_qweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn2to_kweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn2to_vweight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn2to_out0weight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0attn2to_out0bias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0norm1weight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0norm1bias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0norm2weight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0norm2bias", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0norm3weight", "model_ema.diffusion_modeloutput_blocks111transformer_blocks0norm3bias", "model_ema.diffusion_modeloutput_blocks111proj_outweight", "model_ema.diffusion_modeloutput_blocks111proj_outbias", "model_ema.diffusion_modelout0weight", "model_ema.diffusion_modelout0bias", "model_ema.diffusion_modelout2weight", "model_ema.diffusion_modelout2bias".
I reloaded the UI, noticed the VAE changed back to auto, tried to reload the custom one, and got "RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0!"
Am I doing something wrong or missing a step?
Greetings and thanks!
Beta Was this translation helpful? Give feedback.
All reactions