Method
__init__
(self,
num_channels: int,
flip_sin_to_cos: bool,
downscale_freq_shift: float,
scale: int = 1)
Source from the content-addressed store, hash-verified
| 543 | class Timesteps(Module): |
| 544 | |
| 545 | def __init__(self, |
| 546 | num_channels: int, |
| 547 | flip_sin_to_cos: bool, |
| 548 | downscale_freq_shift: float, |
| 549 | scale: int = 1): |
| 550 | super().__init__() |
| 551 | self.num_channels = num_channels |
| 552 | self.flip_sin_to_cos = flip_sin_to_cos |
| 553 | self.downscale_freq_shift = downscale_freq_shift |
| 554 | self.scale = scale |
| 555 | |
| 556 | def forward(self, timesteps) -> Tensor: |
| 557 | t_emb = get_timestep_embedding( |
Callers
nothing calls this directly
Tested by
no test coverage detected