From a1c50b6ff8c322e065a6e2c80c4d80e92fd2da69 Mon Sep 17 00:00:00 2001
From: Yousef Rafat <81116377+yousef-rafat@users.noreply.github.com>
Date: Thu, 31 Jul 2025 22:28:56 +0300
Subject: [PATCH] device error fix

---
 comfy/ldm/hunyuan3dv2_1/hunyuandit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/ldm/hunyuan3dv2_1/hunyuandit.py b/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
index 2c71690ff..7c33d0aaa 100644
--- a/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
+++ b/comfy/ldm/hunyuan3dv2_1/hunyuandit.py
@@ -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: