ByteMorpher-Demo

Paused

Boese0601 commited on May 23

Commit

0e6f992

verified ·

1 Parent(s): 40c798c

Update src/flux/xflux_pipeline.py

Files changed (1) hide show

src/flux/xflux_pipeline.py CHANGED Viewed

@@ -293,6 +293,7 @@ class XFluxPipeline:
             1, height, width, device=self.device,
             dtype=torch.bfloat16, seed=seed
         )
         timesteps = get_schedule(
             num_steps,
             (width // 8) * (height // 8) // (16 * 16),

             1, height, width, device=self.device,
             dtype=torch.bfloat16, seed=seed
         )
+        print("x shape:", x.shape)
         timesteps = get_schedule(
             num_steps,
             (width // 8) * (height // 8) // (16 * 16),