From c46a4bbefdaa08f2724146c4fd7e93532bbdffac Mon Sep 17 00:00:00 2001 From: Carl Worth Date: Wed, 7 Dec 2022 11:52:44 -0800 Subject: [PATCH] Switch to new 2.1 model of Stable Diffusion and also 768x768 pixels That's a slightly more expensive mode to run in, but should still give up to 1000 images for only $10. --- generate-image.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/generate-image.py b/generate-image.py index 3a2d3f3..408fe21 100755 --- a/generate-image.py +++ b/generate-image.py @@ -63,15 +63,17 @@ def main() -> None: stability_api = client.StabilityInference( key=os.environ['STABILITY_KEY'], verbose=True, - engine="stable-diffusion-512-v2-0" + engine="stable-diffusion-768-v2-1" ) prompt_string = " ".join(args.prompt) if args.seed: - answers = stability_api.generate(seed=args.seed, prompt=prompt_string) + answers = stability_api.generate(width=768, height=768, + seed=args.seed, prompt=prompt_string) else: - answers = stability_api.generate(prompt=prompt_string) + answers = stability_api.generate(width=768, height=768, + prompt=prompt_string) results = [] for resp in answers: -- 2.43.0