↓ 3 callersFunctionunet_downblock_to_diffusers_checkpoint(
model, checkpoint, *, diffusers_down_block_idx, original_down_block_idx, original_unet_prefix, num_head_
diffusers/scripts/convert_kakao_brain_unclip_to_diffusers.py:626
↓ 3 callersFunctionunet_upblock_to_diffusers_checkpoint(
model, checkpoint, *, diffusers_up_block_idx, original_up_block_idx, original_unet_prefix, num_head_chan
diffusers/scripts/convert_kakao_brain_unclip_to_diffusers.py:729
↓ 2 callersMethod__init__(self, normalized_shape, eps=1e-05, weight=True, dtype=None, device=None)
llava/model/language_model/mpt/norm.py:35
↓ 2 callersFunction_bwd_kernel_one_col_block(start_n, Q, K, V, Bias, DO, DQ, DK, DV, LSE, D, softmax_scale, stride_qm, stride_kn, stride_vn, stride_bm, st
llava/model/language_model/mpt/flash_attn_triton.py:184
↓ 2 callersFunction_bwd_store_dk_dv(dk_ptrs, dv_ptrs, dk, dv, offs_n, offs_d, seqlen_k, headdim, EVEN_M: tl.constexpr, EVEN_N: tl.constexpr, EVEN
llava/model/language_model/mpt/flash_attn_triton.py:168