Spaces:

tencent
/

Hunyuan3D-2mv

Running on Zero

ZeqiangLai commited on 1 day ago

Commit

1fd9a2d

verified ·

1 Parent(s): cc9db82

Update hy3dgen/shapegen/pipelines.py

Files changed (1) hide show

hy3dgen/shapegen/pipelines.py CHANGED Viewed

@@ -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)
             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)
                 def cat_recursive(a, b):
                     if isinstance(a, torch.Tensor):

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