Skip to content

Commit be5d742

Browse files
committed
conditioning additions inline with CombinedTimestepGuidanceTextProjEmbeddings
1 parent 5e181ed commit be5d742

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/models/transformers/transformer_hunyuan_video.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,6 @@ def forward(
293293
timesteps_proj = self.time_proj(timestep)
294294
timesteps_emb = self.timestep_embedder(timesteps_proj.to(dtype=pooled_projection.dtype)) # (N, D)
295295
pooled_projections = self.text_embedder(pooled_projection)
296-
conditioning = timesteps_emb + pooled_projections
297296

298297
token_replace_emb = None
299298
if self.image_condition_type == "token_replace":
@@ -305,8 +304,9 @@ def forward(
305304
if self.guidance_embedder is not None:
306305
guidance_proj = self.time_proj(guidance)
307306
guidance_emb = self.guidance_embedder(guidance_proj.to(dtype=pooled_projection.dtype))
308-
conditioning = conditioning + guidance_emb
309-
307+
conditioning = timesteps_emb + guidance_emb + pooled_projections
308+
else:
309+
conditioning = timesteps_emb + pooled_projections
310310
return conditioning, token_replace_emb
311311

312312

0 commit comments

Comments
 (0)