(self, t, dtype=torch.float32)
| 227 | return embedding |
| 228 | |
| 229 | def forward(self, t, dtype=torch.float32): |
| 230 | t_freq = self.timestep_embedding(t, self.frequency_embedding_size).to(dtype) |
| 231 | t_emb = self.mlp(t_freq) |
| 232 | return t_emb |
| 233 | |
| 234 | |
| 235 | class FinalLayer(nn.Module): |
nothing calls this directly
no test coverage detected