Update src/flux/xflux_pipeline.py

src/flux/xflux_pipeline.py

@@ -285,7 +285,7 @@ class XFluxPipeline:
         neg_image_proj=None,
         ip_scale=1.0,
         neg_ip_scale=1.0,
-        spatial_condition=False,
+        spatial_condition=True,
         source_image=None,
         share_position_embedding=False
     ):
@@ -312,7 +312,7 @@ 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: