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

fix test

Browse Source
This commit is contained in:
Qing 2023-03-27 20:54:05 +08:00
parent 2e6e87275e
commit 8af5df491e
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lama_cleaner/tests/test_model.py
View File

@ -132,16 +132,20 @@ def test_zits(strategy, zits_wireframe):
)
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL], "no_half", [True, False])
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
@pytest.mark.parametrize("no_half", [True, False])
def test_mat(strategy, no_half):
model = ModelManager(name="mat", device=device, no_half=no_half)
cfg = get_config(strategy)
for _ in range(10):
assert_equal(
model, cfg, f"mat_{strategy.capitalize()}_result.png",
model,
cfg,
f"mat_{strategy.capitalize()}_result.png",
)
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
def test_fcf(strategy):
model = ModelManager(name="fcf", device=device)
@ -188,4 +192,3 @@ def test_manga(strategy):
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)
>>>>>>> 944494a (mat support float16)