novelai-python 0.3.6__tar.gz → 0.3.7__tar.gz

{novelai_python-0.3.6 → novelai_python-0.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: novelai-python
-Version: 0.3.6
+Version: 0.3.7
 Summary: Novelai Python Binding With Pydantic
 Author-Email: sudoskys <coldlando@hotmail.com>
 License: Apache License 2.0

{novelai_python-0.3.6 → novelai_python-0.3.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "novelai-python"
-version = "0.3.6"
+version = "0.3.7"
 description = "Novelai Python Binding With Pydantic"
 authors = [
     { name = "sudoskys", email = "coldlando@hotmail.com" },

{novelai_python-0.3.6 → novelai_python-0.3.7}/src/novelai_python/sdk/ai/generate_image/__init__.py

@@ -146,7 +146,7 @@ class GenerateImageInfer(ApiBaseModel):
             return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
         @staticmethod
-        def add_image_to_black_background(image: Union[str, bytes], width: int, height: int):
+        def add_image_to_black_background(image: Union[str, bytes], width: int = 448, height: int = 448):
             """
             将图像缩放到一个指定的黑背景上，使其最大，比例不变。
             :param image: 图像
@@ -201,7 +201,7 @@ class GenerateImageInfer(ApiBaseModel):
             if self.image is not None:
                 self.image = self.resize_image(self.image, self.width, self.height)
             if self.reference_image is not None:
-                self.reference_image = self.add_image_to_black_background(self.reference_image, self.width, self.height)
+                self.reference_image = self.add_image_to_black_background(self.reference_image, width=448, height=448)
             return self
 
         @field_validator('width')