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

device error fix

Browse Source
This commit is contained in:
Yousef Rafat 2025-07-31 22:28:56 +03:00
parent c39c3ffece
commit a1c50b6ff8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/hunyuan3dv2_1/hunyuandit.py
View File

@ -80,7 +80,7 @@ class MoEGate(nn.Module):
self.alpha = aux_loss_alpha
self.gating_dim = embed_dim
self.weight = nn.Parameter(torch.empty((self.n_routed_experts, self.gating_dim)), device = device, dtype = dtype)
self.weight = nn.Parameter(torch.empty((self.n_routed_experts, self.gating_dim), device = device, dtype = dtype))
def forward(self, hidden_states: torch.Tensor) -> torch.Tensor: