Method forward

models/pos_embed.py:139–146

(self, t)

    @torch.compiler.disable
    def forward(self, t):
        
        cos = self.freqs_cos.clone().to(t.dtype)
        sin = self.freqs_sin.clone().to(t.dtype)
        

        results =  t * cos + rotate_half(t) * sin
        return results.contiguous()