| 414 | |
| 415 | |
| 416 | class Timestep(nn.Module): |
| 417 | def __init__(self, dim): |
| 418 | super().__init__() |
| 419 | self.dim = dim |
| 420 | |
| 421 | def forward(self, t): |
| 422 | return timestep_embedding(t, self.dim) |
| 423 | |
| 424 | |
| 425 | class UNetModel(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected