code: def attention_to_diffusers_checkpoint(
      checkpoint,
      *,
      diffusers_attention_prefix,
      attention_prefix,
      num_head_channels,
  ):
      diffusers_checkpoint = {}
  
      # <original>.norm -> <diffusers>.group_norm
      diffusers_checkpoint.update(
          {
              f"{diffusers_attention_prefix}.group_norm.weight": checkpoint[
                  f"{attention_prefix}.norm.weight"
              ],
              f"{diffusers_attention_prefix}.group_norm.bias": checkpoint[
                  f"{attention_prefix}.norm.bias"
              ],
          }
      )
  
      # <original>.qkv -> <diffusers>.{query, key, value}
      [q_weight, k_weight, v_weight], [
          q_bias,
          k_bias,
          v_bias,
      ] = split_attentions(
          weight=checkpoint[f"{attention_prefix}.qkv.weight"][:, :, 0],
          bias=checkpoint[f"{attention_prefix}.qkv.bias"],
          split=3,
          chunk_size=num_head_channels,
      )
  
      diffusers_checkpoint.update(
          {
              f"{diffusers_attention_prefix}.to_q.weight": q_weight,
              f"{diffusers_attention_prefix}.to_q.bias": q_bias,
              f"{diffusers_attention_prefix}.to_k.weight": k_weight,
              f"{diffusers_attention_prefix}.to_k.bias": k_bias,
              f"{diffusers_attention_prefix}.to_v.weight": v_weight,
              f"{diffusers_attention_prefix}.to_v.bias": v_bias,
          }
      )
  
      # <original>.encoder_kv -> <diffusers>.{context_key, context_value}
      [encoder_k_weight, encoder_v_weight], [
          encoder_k_bias,
          encoder_v_bias,
      ] = split_attentions(
          weight=checkpoint[f"{attention_prefix}.encoder_kv.weight"][
              :, :, 0
          ],
          bias=checkpoint[f"{attention_prefix}.encoder_kv.bias"],
          split=2,
          chunk_size=num_head_channels,
      )
  
      diffusers_checkpoint.update(
          {
              f"{diffusers_attention_prefix}.add_k_proj.weight": encoder_k_weight,
              f"{diffusers_attention_prefix}.add_k_proj.bias": encoder_k_bias,
              f"{diffusers_attention_prefix}.add_v_proj.weight": encoder_v_weight,
              f"{diffusers_attention_prefix}.add_v_proj.bias": encoder_v_bias,
          }
      )
  
      # <original>.proj_out (1d conv) -> <diffusers>.proj_attn (linear)
      diffusers_checkpoint.update(
          {
              f"{diffusers_attention_prefix}.to_out.0.weight": checkpoint[
                  f"{attention_prefix}.proj_out.weight"
              ][
                  :, :, 0
              ],
              f"{diffusers_attention_prefix}.to_out.0.bias": checkpoint[
                  f"{attention_prefix}.proj_out.bias"
              ],
          }
      )
  
      return diffusers_checkpoint
  
imports: import argparse
  import os
  import tempfile
  import torch
  from accelerate import load_checkpoint_and_dispatch
  from diffusers import UNet2DConditionModel
  from diffusers.models.prior_transformer import PriorTransformer
  from diffusers.models.vq_model import VQModel
label_desc: Use image node_id 19 for calling a global function with example usage: attention_to_diffusers_checkpoint(checkpoint) and returns: diffusers_checkpoint