Update src/flux/sampling.py

src/flux/sampling.py

@@ -168,8 +168,6 @@ def denoise(
     for t_curr, t_prev in zip(timesteps[:-1], timesteps[1:]):
         t_vec = torch.full((img.shape[0],), t_curr, dtype=img.dtype, device=img.device)
         if source_image is not None:  # spatial condition or refnet
-            print("source_image shape", source_image.shape)
-            print("img shape", img.shape)
             img = torch.cat([source_image, img],dim=-2)
         pred = model(
             img=img,