(self, timesteps)
| 151 | self.downscale_freq_shift = downscale_freq_shift |
| 152 | |
| 153 | def forward(self, timesteps): |
| 154 | t_emb = get_timestep_embedding( |
| 155 | timesteps, |
| 156 | self.num_channels, |
| 157 | flip_sin_to_cos=self.flip_sin_to_cos, |
| 158 | downscale_freq_shift=self.downscale_freq_shift, |
| 159 | ) |
| 160 | return t_emb |
| 161 | |
| 162 | |
| 163 | class TimestepEmbedding(nn.Module): |
nothing calls this directly
no test coverage detected