↓ 3 callersFunctionpatch_pipe(
pipe,
maybe_unet_path,
token: Optional[str] = None,
r: int = 4,
patch_unet=True,
pat
lora_diffusion/lora.py:979
↓ 2 callersMethod__init__(
self, unified_label_file, dataset_name, cfg,
distributed, output_dir=None)
UniDet_eval/experts/obj_detection/unidet/evaluation/multi_dataset_evaluator.py:345
↓ 2 callersMethod__init__(self, normalized_shape, eps=1e-05, weight=True, dtype=None, device=None)
MLLM_eval/ShareGPT4V-CoT_eval/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
MLLM_eval/ShareGPT4V-CoT_eval/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
MLLM_eval/ShareGPT4V-CoT_eval/llava/model/language_model/mpt/flash_attn_triton.py:168
↓ 2 callersFunction_normal_param_init_fn_(module: nn.Module, std: float, n_layers: int, d_model: Optional[int]=None, init_div_is_residual: Union[int, f
MLLM_eval/ShareGPT4V-CoT_eval/llava/model/language_model/mpt/param_init_fns.py:124
↓ 2 callersFunctiongenerate(model, input_ids, do_sample, temperature, top_p, max_new_tokens, streamer, stopping_criteria, image_args, sto
MLLM_eval/ShareGPT4V-CoT_eval/Share_eval.py:55
↓ 2 callersFunctionloss_step(
batch,
unet,
vae,
text_encoder,
scheduler,
train_inpainting=False,
t_mutliplier=
GORS_finetune/lora_diffusion/cli_lora_pti.py:251