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

add sdxl

Browse Source
This commit is contained in:
Qing
2023-11-14 14:19:56 +08:00
parent 78c8d8dbdb
commit 8fbc8059e1
7 changed files with 246 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lama_cleaner/tests/test_outpainting.py
View File

@@ -57,7 +57,7 @@ def test_outpainting(name, sd_device, rect):
croper_y=rect[1],
croper_width=rect[2],
croper_height=rect[3],
sd_guidance_scale=4,
sd_guidance_scale=8.0,
sd_sampler=SDSampler.dpm_plus_plus,
)
@@ -75,7 +75,7 @@ def test_outpainting(name, sd_device, rect):
@pytest.mark.parametrize(
"rect",
[
[-100, -100, 768, 768],
[-128, -128, 768, 768],
],
)
def test_kandinsky_outpainting(name, sd_device, rect):
@@ -86,7 +86,7 @@ def test_kandinsky_outpainting(name, sd_device, rect):
return
model = ModelManager(
name=name,
name="sd1.5",
device=torch.device(sd_device),
hf_access_token="",
sd_run_local=True,
@@ -105,7 +105,7 @@ def test_kandinsky_outpainting(name, sd_device, rect):
croper_y=rect[1],
croper_width=rect[2],
croper_height=rect[3],
sd_guidance_scale=4,
sd_guidance_scale=7,
sd_sampler=SDSampler.dpm_plus_plus,
)
@@ -115,4 +115,6 @@ def test_kandinsky_outpainting(name, sd_device, rect):
f"{name.replace('.', '_')}_outpainting_dpm++_{'_'.join(map(str, rect))}.png",
img_p=current_dir / "cat.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
fx=1,
fy=1,
)