emmi_inference.models.modules.rope

Functions

rope(x, freqs)

RoPE via polar coordinate rotations.

Module Contents

emmi_inference.models.modules.rope.rope(x, freqs)

RoPE via polar coordinate rotations.

Parameters:
  • x (torch.Tensor)

  • freqs (torch.Tensor)

Return type:

torch.Tensor