Update hy3dgen/shapegen/pipelines.py

hy3dgen/shapegen/pipelines.py

@@ -431,7 +431,7 @@ class Hunyuan3DDiTPipeline:
         cond = self.conditioner(image=image, **additional_cond_inputs)
 
         if do_classifier_free_guidance:
-            un_cond = self.conditioner.unconditional_embedding(bsz)
+            un_cond = self.conditioner.unconditional_embedding(bsz, **additional_cond_inputs)
 
             if dual_guidance:
                 un_cond_drop_main = copy.deepcopy(un_cond)
@@ -447,7 +447,7 @@ class Hunyuan3DDiTPipeline:
 
                 cond = cat_recursive(cond, un_cond_drop_main, un_cond)
             else:
-                un_cond = self.conditioner.unconditional_embedding(bsz)
+                un_cond = self.conditioner.unconditional_embedding(bsz, **additional_cond_inputs)
 
                 def cat_recursive(a, b):
                     if isinstance(a, torch.Tensor):