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

better handle scan single file diffusion

Browse Source
This commit is contained in:
Qing
2024-04-12 18:52:33 +08:00
parent 0a262fa811
commit 76823355fe
3 changed files with 7 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
iopaint/tests/test_brushnet.py
View File

@@ -42,7 +42,7 @@ def test_runway_brushnet(device, sampler):
assert_equal(
model,
cfg,
f"brushnet_runway_1_5_freeu_device_{device}.png",
f"brushnet_random_mask_{device}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)
@@ -74,16 +74,13 @@ def test_brushnet_local_file_path(device, sampler, name):
brushnet_method=SD_BRUSHNET_CHOICES[1]
)
cfg.sd_sampler = sampler
name = f"device_{device}_{sampler}_{name}"
is_sdxl = "sd_xl" in name
assert_equal(
model,
cfg,
f"brushnet_sd_local_model_{name}.png",
f"brushnet_segmentation_mask_{device}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
fx=1.5 if is_sdxl else 1,
fy=1.5 if is_sdxl else 1,
fx=1,
fy=1,
)