↓ 27 callersMethodpad(
self,
x: torch.Tensor,
seq_dim: int,
value: Union[int, float],
lengt
olmo/nn/cp_load_balancer.py:323
↓ 15 callersMethod__init__(self, dim: int, hidden_dim: int, hidden_act: str, device: Union[str, torch.device] = None)
olmo/hf_model/modeling_molmo2.py:87
↓ 7 callersFunctionbatch_pixels_to_patchesReshape images of [n_images, h, w, 3] -> [n_images, n_patches, pixels_per_patch]
olmo/preprocessing/preprocessor_utils.py:111