Update defaults to SDXL v0.9 (#245)

LICENSE

@@ -1,5 +1,5 @@
 
-    Copyright (c) Stability.ai, All rights reserved.
+    Copyright (c) Stability AI Ltd, All rights reserved.
 
     Permission is hereby granted, free of charge, to any person obtaining a copy
     of this software and associated documentation files (the "Software"), to deal

README.md

@@ -26,7 +26,7 @@ Install the [PyPI](https://pypi.org/project/stability-sdk/) package via:
 
 Then to invoke:
 
-`python3 -m stability_sdk generate -W 512 -H 512 "A stunning house."`
+`python3 -m stability_sdk generate -W 1024 -H 1024 "A stunning house."`
 
 It will generate and put PNGs in your current directory.
 
@@ -61,9 +61,9 @@ positional arguments:
 options:
   -h, --help            show this help message and exit
   --height HEIGHT, -H HEIGHT
-                        [512] height of image
+                        [1024] height of image
   --width WIDTH, -W WIDTH
-                        [512] width of image
+                        [1024] width of image
   --start_schedule START_SCHEDULE
                         [0.5] start schedule for init image (must be greater than 0; 1 is full strength
                         text prompt, no trace of image)

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name='stability-sdk',
-    version='0.8.1',
+    version='0.8.2',
     author='Stability AI',
     author_email='[email protected]',
     maintainer='Stability AI',

src/stability_sdk/animation.py

@@ -62,7 +62,7 @@ class BasicSettings(param.Parameterized):
         check_on_set=False, # allow old and new models without raising ValueError
         objects=[
             "stable-diffusion-v1-5", "stable-diffusion-512-v2-1", "stable-diffusion-768-v2-1", 
-            "stable-diffusion-depth-v2-0", "stable-diffusion-xl-beta-v2-2-2",
+            "stable-diffusion-depth-v2-0", "stable-diffusion-xl-beta-v2-2-2", "stable-diffusion-xl-1024-v0-9",
             "custom"
         ]
     )

src/stability_sdk/api.py

@@ -69,7 +69,7 @@ def __init__(
             host: str="", 
             api_key: str=None, 
             stub: generation_grpc.GenerationServiceStub=None,
-            generate_engine_id: str="stable-diffusion-xl-beta-v2-2-2",
+            generate_engine_id: str="stable-diffusion-xl-1024-v0-9",
             inpaint_engine_id: str="stable-inpainting-512-v2-0",
             interpolate_engine_id: str="interpolation-server-v1",
             transform_engine_id: str="transform-server-v1",
@@ -104,8 +104,8 @@ def generate(
         self,
         prompts: List[str], 
         weights: List[float], 
-        width: int = 512, 
-        height: int = 512, 
+        width: int = 1024, 
+        height: int = 1024, 
         steps: Optional[int] = None,
         seed: Union[Sequence[int], int] = 0,
         samples: int = 1,

src/stability_sdk/client.py

@@ -95,7 +95,7 @@ def __init__(
         self,
         host: str = "grpc.stability.ai:443",
         key: str = "",
-        engine: str = "stable-diffusion-xl-beta-v2-2-2",
+        engine: str = "stable-diffusion-xl-1024-v0-9",
         upscale_engine: str = "esrgan-v1-x2plus",
         verbose: bool = False,
         wait_for_ready: bool = True,
@@ -155,8 +155,8 @@ def generate(
         prompt: Union[str, List[str], generation.Prompt, List[generation.Prompt]],
         init_image: Optional[Image.Image] = None,
         mask_image: Optional[Image.Image] = None,
-        height: int = 512,
-        width: int = 512,
+        height: int = 1024,
+        width: int = 1024,
         start_schedule: float = 1.0,
         end_schedule: float = 0.01,
         cfg_scale: float = 7.0,
@@ -493,10 +493,10 @@ def process_cli(logger: logging.Logger = None,
 
     parser_generate = subparsers.add_parser('generate')
     parser_generate.add_argument(
-        "--height", "-H", type=int, default=512, help="[512] height of image"
+        "--height", "-H", type=int, default=1024, help="[1024] height of image"
     )
     parser_generate.add_argument(
-        "--width", "-W", type=int, default=512, help="[512] width of image"
+        "--width", "-W", type=int, default=1024, help="[1024] width of image"
     )
     parser_generate.add_argument(
         "--start_schedule",
@@ -550,7 +550,7 @@ def process_cli(logger: logging.Logger = None,
         "-e",
         type=str,
         help="engine to use for inference",
-        default="stable-diffusion-xl-beta-v2-2-2",
+        default="stable-diffusion-xl-1024-v0-9",
     )
     parser_generate.add_argument(
         "--init_image",

tests/specs.md

@@ -8,7 +8,7 @@ These examples generate and put PNGs in your current directory.
 
 Command line:
 
-`python3 -m stability_sdk generate -W 512 -H 512 "A stunning house."`
+`python3 -m stability_sdk generate -W 1024 -H 1024 "A stunning house."`
 
 SDK Usage: