(self, t)
| 268 | return embedding |
| 269 | |
| 270 | def forward(self, t): |
| 271 | t_freq = self.timestep_embedding( |
| 272 | t, self.frequency_embedding_size, dtype=self.dtype |
| 273 | ) |
| 274 | t_emb = self.mlp(t_freq.to(dtype=self.dtype)) |
| 275 | return t_emb |
| 276 | |
| 277 | |
| 278 | class LabelEmbedder(nn.Module): |
nothing calls this directly
no test coverage detected