Zero out conditioning when there is no prompt #1628

Acly · Mar 9, 2025 · 4c51a38 · 4c51a38
1 parent efb391e
commit 4c51a38
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ai_diffusion/comfy_workflow.py b/ai_diffusion/comfy_workflow.py
@@ -541,6 +541,9 @@ def conditioning_step_range(self, conditioning: Output, range: tuple[float, floa
             end=range[1],
         )
 
+    def conditioning_zero_out(self, conditioning: Output):
+        return self.add("ConditioningZeroOut", 1, conditioning=conditioning)
+
     def instruct_pix_to_pix_conditioning(
         self, positive: Output, negative: Output, vae: Output, pixels: Output
     ):

diff --git a/ai_diffusion/workflow.py b/ai_diffusion/workflow.py
@@ -268,6 +268,8 @@ def encode(self, w: ComfyWorkflow, clip: Output, style_prompt: str | None = None
             if text and self.language:
                 text = w.translate(text)
             self._output = w.clip_text_encode(clip, text)
+            if text == "":
+                self._output = w.conditioning_zero_out(self._output)
             self._clip = clip
         return self._output