↓ 8 callersFunctionload_pretrained_model(model_path, model_base, model_name, load_8bit=False, load_4bit=False, device_map="auto", device="cuda", **kwa
mPLUG-Owl2/mplug_owl2/model/builder.py:26
↓ 4 callersFunctionrepeat_kv This is the equivalent of torch.repeat_interleave(x, dim=1, repeats=n_rep). The hidden states go from (batch, num_key_value_heads, seqlen, he
mPLUG-Owl2/mplug_owl2/model/modeling_llama2.py:22
↓ 2 callersMethod__init__(
self, vision_config=None, visual_abstractor_config=None, text_config=None, num_query_tokens=64, **kw
mPLUG-Owl/mplug_owl_video/configuration_mplug_owl.py:223
↓ 2 callersMethod__init__(
self, vision_config=None, visual_abstractor_config=None, text_config=None, num_query_tokens=64, **kw
mPLUG-Owl/mplug_owl/configuration_mplug_owl.py:225