(self, x)
| 202 | self.register_buffer('pe', pe) |
| 203 | |
| 204 | def forward(self, x): |
| 205 | # not used in the final model |
| 206 | x = x + self.pe[:x.shape[0], :] |
| 207 | return self.dropout(x) |
| 208 | |
| 209 | |
| 210 | class TimestepEmbedder(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected