diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-07 09:09:46 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-07 09:09:46 +0200 |
| commit | d952d467d31786f4a85cc4cb009934cd4ebbba71 (patch) | |
| tree | 68c3f597a86ef3b98734d80cc783aa1f42fe1a41 /infer.py | |
| parent | Update (diff) | |
| download | textual-inversion-diff-d952d467d31786f4a85cc4cb009934cd4ebbba71.tar.gz textual-inversion-diff-d952d467d31786f4a85cc4cb009934cd4ebbba71.tar.bz2 textual-inversion-diff-d952d467d31786f4a85cc4cb009934cd4ebbba71.zip | |
Update
Diffstat (limited to 'infer.py')
| -rw-r--r-- | infer.py | 8 |
1 files changed, 4 insertions, 4 deletions
| @@ -67,8 +67,8 @@ default_cmds = { | |||
| 67 | "batch_num": 1, | 67 | "batch_num": 1, |
| 68 | "steps": 30, | 68 | "steps": 30, |
| 69 | "guidance_scale": 7.0, | 69 | "guidance_scale": 7.0, |
| 70 | "sag_scale": 0.75, | 70 | "sag_scale": 0, |
| 71 | "lora_scale": 0.5, | 71 | "brightness_offset": 0, |
| 72 | "seed": None, | 72 | "seed": None, |
| 73 | "config": None, | 73 | "config": None, |
| 74 | } | 74 | } |
| @@ -192,7 +192,7 @@ def create_cmd_parser(): | |||
| 192 | type=float, | 192 | type=float, |
| 193 | ) | 193 | ) |
| 194 | parser.add_argument( | 194 | parser.add_argument( |
| 195 | "--lora_scale", | 195 | "--brightness_offset", |
| 196 | type=float, | 196 | type=float, |
| 197 | ) | 197 | ) |
| 198 | parser.add_argument( | 198 | parser.add_argument( |
| @@ -392,7 +392,7 @@ def generate(output_dir: Path, pipeline, args): | |||
| 392 | generator=generator, | 392 | generator=generator, |
| 393 | image=init_image, | 393 | image=init_image, |
| 394 | strength=args.image_noise, | 394 | strength=args.image_noise, |
| 395 | # cross_attention_kwargs={"scale": args.lora_scale}, | 395 | brightness_offset=args.brightness_offset, |
| 396 | ).images | 396 | ).images |
| 397 | 397 | ||
| 398 | for j, image in enumerate(images): | 398 | for j, image in enumerate(images): |