Method
__init__
(self, num_channels: int, flip_sin_to_cos: bool, downscale_freq_shift: float)
Source from the content-addressed store, hash-verified
| 145 | |
| 146 | class Timesteps(nn.Module): |
| 147 | def __init__(self, num_channels: int, flip_sin_to_cos: bool, downscale_freq_shift: float): |
| 148 | super().__init__() |
| 149 | self.num_channels = num_channels |
| 150 | self.flip_sin_to_cos = flip_sin_to_cos |
| 151 | self.downscale_freq_shift = downscale_freq_shift |
| 152 | |
| 153 | def forward(self, timesteps): |
| 154 | t_emb = get_timestep_embedding( |
Callers
nothing calls this directly
Tested by
no test coverage detected