fixed bug in defining embed dim

SammyAgrawal · SammyAgrawal · commit f1b3cca0ecea · 2025-08-10T06:13:37.000Z
diff --git a/src/diffusers/models/unets/unet_1d.py b/src/diffusers/models/unets/unet_1d.py
@@ -101,7 +101,7 @@ def __init__(
         # time
         if time_embedding_type == "fourier":
             self.time_proj = GaussianFourierProjection(
-                embedding_size=8, set_W_to_weight=False, log=False, flip_sin_to_cos=flip_sin_to_cos
+                embedding_size=block_out_channels[0], set_W_to_weight=False, log=False, flip_sin_to_cos=flip_sin_to_cos
             )
             timestep_input_dim = 2 * block_out_channels[0]
         elif time_embedding_type == "positional":

Original file line number	Diff line number	Diff line change
`@@ -101,7 +101,7 @@ def __init__(`
`101`	`101`	`# time`
`102`	`102`	`if time_embedding_type == "fourier":`
`103`	`103`	`self.time_proj = GaussianFourierProjection(`
`104`		`- embedding_size=8, set_W_to_weight=False, log=False, flip_sin_to_cos=flip_sin_to_cos`
	`104`	`+ embedding_size=block_out_channels[0], set_W_to_weight=False, log=False, flip_sin_to_cos=flip_sin_to_cos`
`105`	`105`	`)`
`106`	`106`	`timestep_input_dim = 2 * block_out_channels[0]`
`107`	`107`	`elif time_embedding_type == "positional":`