diff options
Diffstat (limited to 'infer.py')
-rw-r--r-- | infer.py | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -28,7 +28,7 @@ default_cmds = { | |||
28 | "prompt": None, | 28 | "prompt": None, |
29 | "negative_prompt": None, | 29 | "negative_prompt": None, |
30 | "image": None, | 30 | "image": None, |
31 | "image_strength": .7, | 31 | "image_strength": .3, |
32 | "width": 512, | 32 | "width": 512, |
33 | "height": 512, | 33 | "height": 512, |
34 | "batch_size": 1, | 34 | "batch_size": 1, |
@@ -225,6 +225,7 @@ def generate(output_dir, pipeline, args): | |||
225 | guidance_scale=args.guidance_scale, | 225 | guidance_scale=args.guidance_scale, |
226 | generator=generator, | 226 | generator=generator, |
227 | latents=init_image, | 227 | latents=init_image, |
228 | strength=args.image_strength, | ||
228 | ).images | 229 | ).images |
229 | 230 | ||
230 | for j, image in enumerate(images): | 231 | for j, image in enumerate(images): |