(self, timesteps)
| 90 | self.computation_device = computation_device |
| 91 | |
| 92 | def forward(self, timesteps): |
| 93 | t_emb = get_timestep_embedding( |
| 94 | timesteps, |
| 95 | self.num_channels, |
| 96 | flip_sin_to_cos=self.flip_sin_to_cos, |
| 97 | downscale_freq_shift=self.downscale_freq_shift, |
| 98 | computation_device=self.computation_device, |
| 99 | ) |
| 100 | return t_emb |
| 101 | |
| 102 | |
| 103 | class TrainableTemporalTimesteps(torch.nn.Module): |
nothing calls this directly
no test coverage detected