summaryrefslogtreecommitdiffstats
path: root/infer.py
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2022-10-02 16:30:30 +0200
committerVolpeon <git@volpeon.ink>2022-10-02 16:30:30 +0200
commit8716116ada220172c36aa34a138f0a0ebefdd684 (patch)
tree0f10e1254ad606c64dd0a4098a2b871fd67e8a18 /infer.py
parentMore consistent euler_a (diff)
downloadtextual-inversion-diff-8716116ada220172c36aa34a138f0a0ebefdd684.tar.gz
textual-inversion-diff-8716116ada220172c36aa34a138f0a0ebefdd684.tar.bz2
textual-inversion-diff-8716116ada220172c36aa34a138f0a0ebefdd684.zip
Fix img2img
Diffstat (limited to 'infer.py')
-rw-r--r--infer.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/infer.py b/infer.py
index c40335c..f2c380f 100644
--- a/infer.py
+++ b/infer.py
@@ -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):