kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

sd add strength

Browse Source
This commit is contained in:
Qing 2023-11-14 14:02:15 +08:00
parent d061b07029
commit 53b9f3e92f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lama_cleaner/model/sd.py
View File

@ -146,6 +146,7 @@ class SD(DiffusionInpaintModel):
negative_prompt=config.negative_prompt, negative_prompt=config.negative_prompt,
mask_image=PIL.Image.fromarray(mask[:, :, -1], mode="L"), mask_image=PIL.Image.fromarray(mask[:, :, -1], mode="L"),
num_inference_steps=config.sd_steps, num_inference_steps=config.sd_steps,
strength=config.sd_strength,
guidance_scale=config.sd_guidance_scale, guidance_scale=config.sd_guidance_scale,
output_type="np", output_type="np",
callback=self.callback, callback=self.callback,