From 077815c4d8071d33c762de0ee14e515a193d98d6 Mon Sep 17 00:00:00 2001 From: George Petterson Date: Thu, 26 Sep 2024 11:09:12 -0500 Subject: [PATCH] Small fix --- apps/shark_studio/api/shark_api.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/apps/shark_studio/api/shark_api.py b/apps/shark_studio/api/shark_api.py index f772169df6..b171bb3d3a 100644 --- a/apps/shark_studio/api/shark_api.py +++ b/apps/shark_studio/api/shark_api.py @@ -121,8 +121,9 @@ def generate_images( pass elif base_model == "sd3": pass - # existing_args = pipeline_args - pass + existing_args = pipeline_args + pipeline = new_pipeline + pipelines[base_model] = (existing_args, pipeline) generated_images = [] for current_batch in range(batch_count):