Small fix

nod-ai · Sep 26, 2024 · 077815c · 077815c
1 parent 38b6248
commit 077815c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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):