(self, timesteps)
| 771 | self.scale = scale |
| 772 | |
| 773 | def forward(self, timesteps): |
| 774 | t_emb = get_timestep_embedding( |
| 775 | timesteps, |
| 776 | self.num_channels, |
| 777 | flip_sin_to_cos=self.flip_sin_to_cos, |
| 778 | downscale_freq_shift=self.downscale_freq_shift, |
| 779 | scale=self.scale, |
| 780 | ) |
| 781 | return t_emb |
| 782 | |
| 783 | |
| 784 | class GaussianFourierProjection(nn.Module): |
nothing calls this directly
no test coverage detected