fix(api): turn alternatives back off for SDXL

ssube · Nov 12, 2023 · 3ffbc00 · 3ffbc00
1 parent 6eb014c
commit 3ffbc00
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 19 deletions.
diff --git a/api/onnx_web/chain/blend_img2img.py b/api/onnx_web/chain/blend_img2img.py
@@ -81,10 +81,11 @@ def run(
                 )
             else:
                 # encode and record alternative prompts outside of LPW
-                prompt_embeds = encode_prompt(
-                    pipe, prompt_pairs, params.batch, params.do_cfg()
-                )
-                pipe.unet.set_prompts(prompt_embeds)
+                if not params.is_xl():
+                    prompt_embeds = encode_prompt(
+                        pipe, prompt_pairs, params.batch, params.do_cfg()
+                    )
+                    pipe.unet.set_prompts(prompt_embeds)
 
                 rng = np.random.RandomState(params.seed)
                 result = pipe(

diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py
@@ -130,10 +130,11 @@ def run(
             )
         else:
             # encode and record alternative prompts outside of LPW
-            prompt_embeds = encode_prompt(
-                pipe, prompt_pairs, params.batch, params.do_cfg()
-            )
-            pipe.unet.set_prompts(prompt_embeds)
+            if not params.is_xl():
+                prompt_embeds = encode_prompt(
+                    pipe, prompt_pairs, params.batch, params.do_cfg()
+                )
+                pipe.unet.set_prompts(prompt_embeds)
 
             rng = np.random.RandomState(params.seed)
             result = pipe(

diff --git a/api/onnx_web/chain/upscale_outpaint.py b/api/onnx_web/chain/upscale_outpaint.py
@@ -99,10 +99,11 @@ def run(
                 )
             else:
                 # encode and record alternative prompts outside of LPW
-                prompt_embeds = encode_prompt(
-                    pipe, prompt_pairs, params.batch, params.do_cfg()
-                )
-                pipe.unet.set_prompts(prompt_embeds)
+                if not params.is_xl():
+                    prompt_embeds = encode_prompt(
+                        pipe, prompt_pairs, params.batch, params.do_cfg()
+                    )
+                    pipe.unet.set_prompts(prompt_embeds)
 
                 rng = np.random.RandomState(params.seed)
                 result = pipe(

diff --git a/api/onnx_web/chain/upscale_stable_diffusion.py b/api/onnx_web/chain/upscale_stable_diffusion.py
@@ -48,13 +48,14 @@ def run(
         )
         generator = torch.manual_seed(params.seed)
 
-        prompt_embeds = encode_prompt(
-            pipeline,
-            prompt_pairs,
-            num_images_per_prompt=params.batch,
-            do_classifier_free_guidance=params.do_cfg(),
-        )
-        pipeline.unet.set_prompts(prompt_embeds)
+        if not params.is_xl():
+            prompt_embeds = encode_prompt(
+                pipeline,
+                prompt_pairs,
+                num_images_per_prompt=params.batch,
+                do_classifier_free_guidance=params.do_cfg(),
+            )
+            pipeline.unet.set_prompts(prompt_embeds)
 
         outputs = []
         for source in sources: