helms commited on
Commit
2a0481e
·
1 Parent(s): ab91c4e

change generation engine

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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-v1", # 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 stable-inpainting-v1-0 stable-inpainting-512-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,