↓ 2 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
olmo/hf_model/modeling_molmo2.py:581
↓ 1 callersMethod__init__(self, rope_theta, dim, cache_prefix, cache: BufferCache, max_len=None, device=None)
olmo/models/molmo_point/modules.py:35
↓ 1 callersMethod_format_video_single_point_trackFormat as: [x1,y1,t1], [x2,y2,t2], ...
olmo/preprocessing/point_formatter.py:787