AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Align HunyuanVideoConditionEmbedding with CombinedTimestepGuidanceTextProjEmbeddings (#12316)

Browse Source 
conditioning additions inline with CombinedTimestepGuidanceTextProjEmbeddings

Co-authored-by: Samu Tamminen <samutamm@users.noreply.github.com>
Co-authored-by: YiYi Xu <yixu310@gmail.com>
This commit is contained in:
Samu Tamminen
2026-01-10 01:51:04 +02:00
committed by GitHub
parent 5794ffffbe
commit df6516a716
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/models/transformers/transformer_hunyuan_video.py
View File

@@ -312,7 +312,6 @@ class HunyuanVideoConditionEmbedding(nn.Module):
timesteps_proj = self.time_proj(timestep)
timesteps_emb = self.timestep_embedder(timesteps_proj.to(dtype=pooled_projection.dtype)) # (N, D)
pooled_projections = self.text_embedder(pooled_projection)
conditioning = timesteps_emb + pooled_projections
token_replace_emb = None
if self.image_condition_type == "token_replace":
@@ -324,8 +323,9 @@ class HunyuanVideoConditionEmbedding(nn.Module):
if self.guidance_embedder is not None:
guidance_proj = self.time_proj(guidance)
guidance_emb = self.guidance_embedder(guidance_proj.to(dtype=pooled_projection.dtype))
conditioning = conditioning + guidance_emb
conditioning = timesteps_emb + guidance_emb + pooled_projections
else:
conditioning = timesteps_emb + pooled_projections
return conditioning, token_replace_emb