xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-14 05:10:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model.py

Browse Source
This commit is contained in:
kijai 2025-11-16 01:00:39 +02:00 committed by comfyanonymous
parent 24d1b6b88a
commit 7378bf6a27
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ldm/hunyuan_video/model.py
View File

@ -367,8 +367,9 @@ class HunyuanVideo(nn.Module):
txt_ids = torch.cat((txt_ids, txt_byt5_ids), dim=1) txt_ids = torch.cat((txt_ids, txt_byt5_ids), dim=1)
#todo vision_in #todo vision_in
if self.cond_type_embedding is not None and vision_states is not None: if vision_states is not None:
txt_vision_states = self.vision_in(vision_states) txt_vision_states = self.vision_in(vision_states)
if self.cond_type_embedding is not None:
cond_emb = self.cond_type_embedding(2 * torch.ones_like(txt_vision_states[:, :, 0], dtype=torch.long, device=txt_vision_states.device)) cond_emb = self.cond_type_embedding(2 * torch.ones_like(txt_vision_states[:, :, 0], dtype=torch.long, device=txt_vision_states.device))
txt_vision_states = txt_vision_states + cond_emb txt_vision_states = txt_vision_states + cond_emb
#print("txt_vision_states shape:", txt_vision_states.shape) #print("txt_vision_states shape:", txt_vision_states.shape)