↓ 6 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
javisgpt/model/qwen2_5_vl/modeling_qwen2_5_vl.py:689
↓ 6 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
javisgpt/model/qwen2_vl/modeling_qwen2_vl.py:478
↓ 4 callersFunctionsmart_pad(x: torch.Tensor, pad_len, dim=0, mode="constant", value=0,
pos:Literal["right", "left", "both"
javisgpt/model/blocks.py:38
↓ 1 callersMethod__init__(self, num_tokens, codebook_dim, decay=0.99, eps=1e-5, kmeans_init=True, codebook_init_path='')
javisgpt/model/beats/quantizer.py:71