Update src/flux/xflux_pipeline.py

src/flux/xflux_pipeline.py

@@ -292,8 +292,7 @@ class XFluxPipeline:
         x = get_noise(
             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),
@@ -312,7 +311,6 @@ class XFluxPipeline:
                 source_image = self.ae.encode(source_image.to(self.device).to(torch.float32))
                 # print("ae source image shape:", source_image.shape)
                 source_image = rearrange(source_image, "b c (h ph) (w pw) -> b (h w) (c ph pw)", ph=2, pw=2).to(inp_cond['img'].dtype)
-                print("rearrange ae source image shape:", source_image.shape)
                 # print("rearrange ae source image shape:", source_image.shape)
                 
             if self.offload: