From edaa5734d0f77db301ba82dd87199132dd695956 Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Tue, 12 Sep 2023 10:19:49 -0500 Subject: [PATCH] Update stencil pipeline with the new img2img arg. --- .../src/pipelines/pipeline_shark_stable_diffusion_stencil.py | 1 + 1 file changed, 1 insertion(+) diff --git a/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_stencil.py b/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_stencil.py index 622ba90231..666cc935a6 100644 --- a/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_stencil.py +++ b/apps/stable_diffusion/src/pipelines/pipeline_shark_stable_diffusion_stencil.py @@ -273,6 +273,7 @@ def generate_images( cpu_scheduling, max_embeddings_multiples, use_stencil, + resample_type, ): # Control Embedding check & conversion # TODO: 1. Change `num_images_per_prompt`.