Fixes

nod-ai · Feb 9, 2024 · 99de56a · 99de56a
1 parent 37feded
commit 99de56a
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 1 deletion.
diff --git a/apps/shark_studio/api/sd.py b/apps/shark_studio/api/sd.py
@@ -299,6 +299,7 @@ def produce_img_latents(
         text_embeddings_numpy = text_embeddings.detach().numpy()
         guidance_scale = torch.Tensor([guidance_scale]).to(self.dtype)
         self.load_submodels(["unet"])
+        control_scale = torch.tensor(1.0, dtype=self.dtype)
         for i, t in tqdm(enumerate(total_timesteps)):
             step_start_time = time.time()
             timestep = torch.tensor([t]).to(self.dtype).detach().numpy()
@@ -337,6 +338,32 @@ def produce_img_latents(
                         timestep,
                         text_embeddings_numpy,
                         guidance_scale,
+                        controlnet_latents[0],
+                        controlnet_latents[1],
+                        controlnet_latents[2],
+                        controlnet_latents[3],
+                        controlnet_latents[4],
+                        controlnet_latents[5],
+                        controlnet_latents[6],
+                        controlnet_latents[7],
+                        controlnet_latents[8],
+                        controlnet_latents[9],
+                        controlnet_latents[10],
+                        controlnet_latents[11],
+                        controlnet_latents[12],
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
+                        control_scale,
                     ],
                 )
             # end_profiling(profile_device)

diff --git a/apps/shark_studio/web/ui/sd.py b/apps/shark_studio/web/ui/sd.py
@@ -469,7 +469,7 @@ def base_model_changed(base_model_id):
                     )
                 with gr.Accordion(
                     label="Controlnet Options",
-                    open=False,
+                    open=True,
                     visible=True,
                 ):
                     preprocessed_hints = gr.State([])