(self, timesteps)
| 554 | self.scale = scale |
| 555 | |
| 556 | def forward(self, timesteps) -> Tensor: |
| 557 | t_emb = get_timestep_embedding( |
| 558 | timesteps, |
| 559 | self.num_channels, |
| 560 | flip_sin_to_cos=self.flip_sin_to_cos, |
| 561 | downscale_freq_shift=self.downscale_freq_shift, |
| 562 | scale=self.scale, |
| 563 | ) |
| 564 | return t_emb |
| 565 | |
| 566 | |
| 567 | class PixArtAlphaTextProjection(Module): |