↓ 15 callersFunctionload_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=False, device_map="auto", device="cuda", **kwa
videollava/model/builder.py:27
↓ 5 callersFunctiongeneric_param_init_fn_(module: nn.Module, init_fn_, n_layers: int, d_model: Optional[int]=None, init_div_is_residual: Union[int, flo
videollava/model/language_model/mpt/param_init_fns.py:28
↓ 4 callersFunctionget_model_output(model, video_processor, tokenizer, video, qs, args)
videollava/eval/video/run_inference_video_qa.py:49
↓ 3 callersFunction_flash_attn_backward(do, q, k, v, o, lse, dq, dk, dv, bias=None, causal=False, softmax_scale=None)
videollava/model/language_model/mpt/flash_attn_triton.py:366
↓ 3 callersFunction_flash_attn_forward(q, k, v, bias=None, causal=False, softmax_scale=None)
videollava/model/language_model/mpt/flash_attn_triton.py:329
↓ 2 callersMethod__init__(self, normalized_shape, eps=1e-05, weight=True, dtype=None, device=None)
videollava/model/language_model/mpt/norm.py:35
↓ 2 callersMethod__init__(self, clip_type=('thermal', 'image', 'video', 'depth', 'audio'), use_temp=True, cache_dir='./cache_dir')
videollava/model/multimodal_encoder/languagebind/__init__.py:55
↓ 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
videollava/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
videollava/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
videollava/model/language_model/mpt/param_init_fns.py:124
↓ 2 callersFunctionbuild_prompt_chatbot(problems, shot_qids, prompt_format, use_caption=False, options=["A", "B", "C", "D", "E"], is_test=False)
scripts/convert_sqa_to_llava_base_prompt.py:221
↓ 2 callersFunctioncreate_one_example(format, question, context, choice, answer, lecture, solution, test_example=True)
scripts/convert_sqa_to_llava_base_prompt.py:106
↓ 2 callersFunctioncreate_one_example_gpt4(format, question, context, choice, answer, lecture, solution, test_example=True)
scripts/convert_sqa_to_llava_base_prompt.py:162