fix aspect ratio issue with style API
This commit is contained in:
+1
-1
@@ -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,
|
||||
|
||||
+1
-1
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user