let5see/IOPaint
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Qing
2023-03-01 21:44:02 +08:00
parent 8e5e4892af
commit 66d3c6e322
7 changed files with 644 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lama_cleaner/model/sd.py
View File

@@ -136,7 +136,7 @@ class SD(DiffusionInpaintModel):
callback=self.callback,
height=img_h,
width=img_w,
generator=torch.manual_seed(config.sd_seed)
generator=torch.manual_seed(config.sd_seed),
).images[0]
output = (output * 255).round().astype("uint8")
@@ -163,6 +163,16 @@ class SD15(SD):
model_id_or_path = "runwayml/stable-diffusion-inpainting"
class Anything4(SD):
name = "anything4"
model_id_or_path = "Sanster/anything-4.0-inpainting"
class RealisticVision14(SD):
name = "realisticVision1.4"
model_id_or_path = "Sanster/Realistic_Vision_V1.4-inpainting"
class SD2(SD):
name = "sd2"
model_id_or_path = "stabilityai/stable-diffusion-2-inpainting"