(self, freq=1e4, F0=1.0, scaling_factor=1.0)
| 409 | |
| 410 | class RoPE3D(RoPE1D): |
| 411 | def __init__(self, freq=1e4, F0=1.0, scaling_factor=1.0): |
| 412 | super(RoPE3D, self).__init__(freq, F0, scaling_factor) |
| 413 | self.position_cache = {} |
| 414 | |
| 415 | def get_mesh_3d(self, rope_positions, bsz): |
| 416 | f, h, w = rope_positions |