feat: Customizable Noise Multiplier (#148) (#149)

This commit is contained in:
szriru
2023-06-20 11:23:53 +09:00
committed by GitHub
parent f10d699d33
commit a724bcb810
3 changed files with 36 additions and 0 deletions

View File

@@ -282,6 +282,11 @@ class AfterDetailerScript(scripts.Script):
return args.ad_cfg_scale
return p.cfg_scale
def get_initial_noise_multiplier(self, p, args: ADetailerArgs) -> float | None:
if args.ad_use_noise_multiplier:
return args.ad_noise_multiplier
return None
def infotext(self, p) -> str:
return create_infotext(
p, p.all_prompts, p.all_seeds, p.all_subseeds, None, 0, 0
@@ -340,6 +345,7 @@ class AfterDetailerScript(scripts.Script):
width, height = self.get_width_height(p, args)
steps = self.get_steps(p, args)
cfg_scale = self.get_cfg_scale(p, args)
initial_noise_multiplier = self.get_initial_noise_multiplier(p, args)
sampler_name = p.sampler_name
if sampler_name in ["PLMS", "UniPC"]:
@@ -355,6 +361,7 @@ class AfterDetailerScript(scripts.Script):
inpaint_full_res=args.ad_inpaint_only_masked,
inpaint_full_res_padding=args.ad_inpaint_only_masked_padding,
inpainting_mask_invert=0,
initial_noise_multiplier=initial_noise_multiplier,
sd_model=p.sd_model,
outpath_samples=p.outpath_samples,
outpath_grids=p.outpath_grids,