Skip to content

Commit 8d415a6

Browse files
authored
PRX Set downscale_freq_shift to 0 for consistency with internal implementation (#12791)
fix timestepembeddings downscale_freq_shift to be consitant with Photoroom's original code
1 parent 7de51b8 commit 8d415a6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/diffusers/models/transformers/transformer_prx.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -694,6 +694,7 @@ def _compute_timestep_embedding(self, timestep: torch.Tensor, dtype: torch.dtype
694694
max_period=self.time_max_period,
695695
scale=self.time_factor,
696696
flip_sin_to_cos=True, # Match original cos, sin order
697+
downscale_freq_shift=0.0,
697698
).to(dtype)
698699
)
699700

0 commit comments

Comments
 (0)