Unverified Commit 8d415a6f authored by David Bertoin's avatar David Bertoin Committed by GitHub
Browse files

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
parent 7de51b82
...@@ -694,6 +694,7 @@ class PRXTransformer2DModel(ModelMixin, ConfigMixin, AttentionMixin): ...@@ -694,6 +694,7 @@ class PRXTransformer2DModel(ModelMixin, ConfigMixin, AttentionMixin):
max_period=self.time_max_period, max_period=self.time_max_period,
scale=self.time_factor, scale=self.time_factor,
flip_sin_to_cos=True, # Match original cos, sin order flip_sin_to_cos=True, # Match original cos, sin order
downscale_freq_shift=0.0,
).to(dtype) ).to(dtype)
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment