↓ 8 callersMethodprepare_vae_images(self, curr_kvlens, curr_rope, images, transforms, new_token_ids, timestep=0)
modeling/bagel/bagel.py:417
↓ 4 callersMethod_add_image(self, data, image, need_loss, need_vae, need_vit, enable_cfg=True)
data/interleave_datasets/interleave_t2i_dataset.py:36
↓ 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
modeling/qwen2/modeling_qwen2.py:205
↓ 3 callersMethodevaluate(self, image_prompts, text_prompt, extract_overall_score_only=False, extract_all_score=True, echo_output=False
eval/gen/gedit/viescore/__init__.py:37