From 8d415a6f481ff1b26168c046267628419650f930 Mon Sep 17 00:00:00 2001 From: David Bertoin Date: Thu, 4 Dec 2025 21:57:14 +0100 Subject: [PATCH] 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 --- src/diffusers/models/transformers/transformer_prx.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/diffusers/models/transformers/transformer_prx.py b/src/diffusers/models/transformers/transformer_prx.py index ccbc83ffca..18ec650bac 100644 --- a/src/diffusers/models/transformers/transformer_prx.py +++ b/src/diffusers/models/transformers/transformer_prx.py @@ -694,6 +694,7 @@ class PRXTransformer2DModel(ModelMixin, ConfigMixin, AttentionMixin): max_period=self.time_max_period, scale=self.time_factor, flip_sin_to_cos=True, # Match original cos, sin order + downscale_freq_shift=0.0, ).to(dtype) )