Spaces:
Runtime error
Runtime error
change generation engine
Browse files
app.py
CHANGED
|
@@ -30,8 +30,9 @@ def infer(prompt):
|
|
| 30 |
stability_api = client.StabilityInference(
|
| 31 |
key=SD_KEY, # AaPI Key reference.
|
| 32 |
verbose=True, # Print debug messages.
|
| 33 |
-
engine="stable-diffusion-
|
| 34 |
-
# Available engines: stable-diffusion-v1 stable-diffusion-v1-5 stable-diffusion-512-v2-0 stable-diffusion-768-v2-0
|
|
|
|
| 35 |
)
|
| 36 |
answers = stability_api.generate(
|
| 37 |
prompt=prompt,
|
|
|
|
| 30 |
stability_api = client.StabilityInference(
|
| 31 |
key=SD_KEY, # AaPI Key reference.
|
| 32 |
verbose=True, # Print debug messages.
|
| 33 |
+
engine="stable-diffusion-xl-beta-v2-2-2", # Set the engine to use for generation.
|
| 34 |
+
# Available engines: stable-diffusion-v1 stable-diffusion-v1-5 stable-diffusion-512-v2-0 stable-diffusion-768-v2-0
|
| 35 |
+
# stable-diffusion-512-v2-1 stable-diffusion-768-v2-1 stable-diffusion-xl-beta-v2-2-2 stable-inpainting-v1-0 stable-inpainting-512-v2-0
|
| 36 |
)
|
| 37 |
answers = stability_api.generate(
|
| 38 |
prompt=prompt,
|