Update stencil pipeline with the new img2img arg.

2026-04-03 03:00:17 -04:00 · 2023-09-12 10:19:49 -05:00
parent 2beb96ad2d
commit edaa5734d0
1 changed files with 1 additions and 0 deletions
--- 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 @@ class StencilPipeline(StableDiffusionPipeline):
        cpu_scheduling,
        max_embeddings_multiples,
        use_stencil,
+        resample_type,
    ):
        # Control Embedding check & conversion
        # TODO: 1. Change `num_images_per_prompt`.