(self, input_ids:Tensor)
| 256 | self.forward_jit = TinyJit(self.forward) |
| 257 | |
| 258 | def forward(self, input_ids:Tensor): |
| 259 | hidden_states = self.backbone(input_ids) |
| 260 | return self.lm_head(hidden_states).realize() |
| 261 | |
| 262 | def __call__(self, input_ids): |
| 263 | return self.forward(input_ids) |