(self, tokens, text, pad_mask)
| 247 | self.pe_shape = pe_shape |
| 248 | |
| 249 | def forward(self, tokens, text, pad_mask): |
| 250 | if self.pe_shape > 30: |
| 251 | tokens = self.loadtoken(tokens, text, pad_mask) |
| 252 | tokens = self.mlp(self.norm(tokens)) |
| 253 | return tokens, self.positional_embedding |
| 254 | |
| 255 | |
| 256 | class CGAttention(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected