Spaces:

Boese0601
/

ByteMorph-Demo

Running on Zero

Boese0601 commited on 20 days ago

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),