Method forward

train/patch/modeling_chatglm.py:130–133

(self, max_seq_len, offset=0)

        return cache

    def forward(self, max_seq_len, offset=0):
        return self.forward_impl(
            max_seq_len, self.dim, dtype=self.inv_freq.dtype, device=self.inv_freq.device
        )


@torch.jit.script