fixes

bghira · Jan 5, 2025 · cba0d88 · cba0d88
1 parent 7509e80
commit cba0d88
Show file tree

Hide file tree

Showing 3 changed files with 2,085 additions and 1,821 deletions.
diff --git a/discord_tron_client/classes/image_manipulation/diffusion.py b/discord_tron_client/classes/image_manipulation/diffusion.py
@@ -391,27 +391,27 @@ def get_pipe(
                 if config.enable_compile() and hasattr(
                     self.pipelines[model_id], "unet"
                 ):
-                    torch._dynamo.config.suppress_errors = True
+                    # torch._dynamo.config.suppress_errors = True
                     self.pipelines[model_id].unet = torch.compile(
                         self.pipelines[model_id].unet,
-                        mode="reduce-overhead",
+                        mode="max-autotune",
                         fullgraph=True,
                     )
                 if config.enable_compile() and hasattr(
                     self.pipelines[model_id], "transformer"
                 ):
-                    torch._dynamo.config.suppress_errors = True
+                    # torch._dynamo.config.suppress_errors = True
                     self.pipelines[model_id].transformer = torch.compile(
                         self.pipelines[model_id].transformer,
-                        mode="reduce-overhead",
+                        mode="max-autotune",
                         fullgraph=True,
                     )
                 if (
                     hasattr(self.pipelines[model_id], "controlnet")
                     and config.enable_compile()
                 ):
                     self.pipelines[model_id].controlnet = torch.compile(
-                        self.pipelines[model_id].controlnet, fullgraph=True
+                        self.pipelines[model_id].controlnet, mode="max-autotune", fullgraph=True
                     )
         else:
             logger.info(f"Keeping existing pipeline. Not creating any new ones.")

diff --git a/discord_tron_client/classes/image_manipulation/pipeline_runners/sd3_runner.py b/discord_tron_client/classes/image_manipulation/pipeline_runners/sd3_runner.py
@@ -53,7 +53,7 @@ def __call__(self, **args):
         if "skip_layer_guidance_end" in args:
             args["skip_layer_guidance_end"] = float(args["skip_layer_guidance_end"])
         if "skip_layer_guidance_scale" in args:
-            args["skip_layer_guidance_end"] = float(args["skip_layer_guidance_scale"])
+            args["skip_layer_guidance_scale"] = float(args["skip_layer_guidance_scale"])
         if "skip_guidance_layers" in args and type(args["skip_guidance_layers"]) == str:
             if args["skip_guidance_layers"].lower() == "none":
                 args["skip_guidance_layers"] = None