more fixes

2025-05-17 20:59:46 +02:00
parent ca9e9b94cb
commit 5ba133cae9
3 changed files with 36 additions and 32 deletions
--- a/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
+++ b/hyvideo/diffusion/pipelines/pipeline_hunyuan_video.py
@@ -1305,7 +1305,7 @@ class HunyuanVideoPipeline(DiffusionPipeline):
                # perform guidance
                if self.do_classifier_free_guidance:
                    if cfg_star_rescale:
-                        batch_size = noise_pred_text.shape[0]
+                        batch_size = 1 
                        positive_flat = noise_pred_text.view(batch_size, -1)
                        negative_flat = noise_pred_uncond.view(batch_size, -1)
                        dot_product = torch.sum(