fix: cor ct ratio inpaint sdxl

R3gm · Jun 16, 2024 · 585eedc · 585eedc
1 parent a622f04
commit 585eedc
Show file tree

Hide file tree

Showing 5 changed files with 425 additions and 423 deletions.
diff --git a/README.md b/README.md
@@ -6,7 +6,7 @@ The goal of this project is to make Stable Diffusion more accessible, simple and
 
 **Installation:**
 ```
-pip install stablepy==0.4.0
+pip install stablepy==0.4.1
 ```
 
 **Usage:**

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "stablepy"
-version = "0.4.0"
+version = "0.4.1"
 description = "A tool for easy use of stable diffusion"
 authors = ["Roger Condori(R3gm) <[email protected]>"]
 readme = "README.md"

diff --git a/stablepy/__version__.py b/stablepy/__version__.py
@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.4.1"
diff --git a/stablepy/diffusers_vanilla/model.py b/stablepy/diffusers_vanilla/model.py
@@ -1959,6 +1959,8 @@ def __call__(
             if self.task_name == "inpaint":
                 pipe_params_config["strength"] = strength
                 pipe_params_config["mask_image"] = control_mask
+                pipe_params_config["height"] = control_image.size[1]
+                pipe_params_config["width"] = control_image.size[0]
             elif self.task_name not in ["txt2img", "inpaint", "img2img"]:
                 if "t2i" not in self.task_name:
                     pipe_params_config[