↓ 7 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
llava/model/language_model/modeling_llama.py:231
↓ 6 callersMethodget_chat_response(self, prompt, temperature=0, max_tokens=256, n=1, patience=10000000, sleep_time=0)
lmms-eval/lmms_eval/tasks/mmrefine/mmrefine_evals.py:48
↓ 5 callersFunctionget_eval_generic(question, answer, pred, task, max_tokens: int, retries: int = 5)
lmms-eval/lmms_eval/tasks/videochatgpt/utils.py:214