From d952d467d31786f4a85cc4cb009934cd4ebbba71 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Fri, 7 Apr 2023 09:09:46 +0200 Subject: Update --- infer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'infer.py') diff --git a/infer.py b/infer.py index 93848d7..8fdf63d 100644 --- a/infer.py +++ b/infer.py @@ -67,8 +67,8 @@ default_cmds = { "batch_num": 1, "steps": 30, "guidance_scale": 7.0, - "sag_scale": 0.75, - "lora_scale": 0.5, + "sag_scale": 0, + "brightness_offset": 0, "seed": None, "config": None, } @@ -192,7 +192,7 @@ def create_cmd_parser(): type=float, ) parser.add_argument( - "--lora_scale", + "--brightness_offset", type=float, ) parser.add_argument( @@ -392,7 +392,7 @@ def generate(output_dir: Path, pipeline, args): generator=generator, image=init_image, strength=args.image_noise, - # cross_attention_kwargs={"scale": args.lora_scale}, + brightness_offset=args.brightness_offset, ).images for j, image in enumerate(images): -- cgit v1.2.3-54-g00ecf