Fixed issue with partial denoising of latents.

2025-04-21 17:12:55 -06:00
parent 8738fe1719
commit 8287164e9f
1 changed files with 11 additions and 8 deletions
@@ -381,14 +381,17 @@ class Flex2Conditioner:
            device='cpu',
            dtype=torch.float32
        )
-
-        out_latent = {
-            "samples": torch.zeros(
-                (batch_size, 16, latent_height, latent_width),
-                device='cpu',
-                dtype=torch.float32
-            )
-        }
+        
+        if latent is not None:
+            out_latent = latent
+        else:
+            out_latent = {
+                "samples": torch.zeros(
+                    (batch_size, 16, latent_height, latent_width),
+                    device='cpu',
+                    dtype=torch.float32
+                )
+            }

        # inpaint
        if inpaint_image is not None: