Update moe.py

fixed not returning y in the forward method of MoEBlock

comfy/ldm/hunyuan3d/model/moe.py

@@ -152,6 +152,8 @@ class MoEBlock(nn.Module):
 
         y = y + self.shared_experts(identity)
 
+        return y
+
     @torch.no_grad()
     def moe_infer(self, x, flat_expert_indices, flat_expert_weights):
         
@@ -200,4 +202,4 @@ def test_moe():
     print(timing)
 
 if __name__ == "__main__":
-    test_moe()
+    test_moe()