From 33fffec3a64ccd6d66aad45764d9d822cda78ad7 Mon Sep 17 00:00:00 2001
From: Mohamad-Stability <mohamad.diab@stability.ai>
Date: Wed, 6 Nov 2024 01:04:03 +0200
Subject: [PATCH] fix aspect ratio issue with style API

---
 __init__.py      | 2 +-
 stability_api.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index a770363..cbbcc97 100644
--- a/__init__.py
+++ b/__init__.py
@@ -11,7 +11,7 @@ NODE_CLASS_MAPPINGS = {
     "Stability Search and Replace": StabilitySearchAndReplace,
     "Stability Outpainting": StabilityOutpainting,
     "Stability SD3": StabilitySD3,
-    "Stability Control Skech": StabilityControlSketch,
+    "Stability Control Sketch": StabilityControlSketch,
     "Stability Control Structure": StabilityControlStructure,
     "Stability Fast Upscale": StabilityFastUpscale,
     "Stability Style": StabilityStyle,
diff --git a/stability_api.py b/stability_api.py
index ac370a1..5144408 100644
--- a/stability_api.py
+++ b/stability_api.py
@@ -52,7 +52,7 @@ class StabilityBase:
 
         image = kwargs.get('image', None)
         if image is not None:
-            if self.API_ENDPOINT not in ["stable-image/control/style", "stable-image/control/sketch", "stable-image/control/structure"]:
+            if self.API_ENDPOINT != "stable-image/control/style":
                 kwargs["mode"] = "image-to-image"
                 kwargs.pop("aspect_ratio", None)
             image = ToPILImage()(image.squeeze(0).permute(2,0,1))