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

update test

Browse Source
This commit is contained in:
Qing 2022-11-04 15:34:40 +08:00
parent b3cb11f804
commit eaef8d143c
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lama_cleaner/tests/test_model.py
View File

@ -160,7 +160,7 @@ def test_fcf(strategy):
)
@pytest.mark.parametrize("sd_device", ['cpu'])
@pytest.mark.parametrize("sd_device", ['cpu', 'cuda'])
@pytest.mark.parametrize("strategy", [HDStrategy.ORIGINAL])
@pytest.mark.parametrize("sampler", [SDSampler.ddim, SDSampler.pndm, SDSampler.k_lms])
@pytest.mark.parametrize("cpu_textencoder", [True, False])
@ -169,6 +169,9 @@ def test_runway_sd_1_5(sd_device, strategy, sampler, cpu_textencoder, disable_ns
def callback(i, t, latents):
print(f"sd_step_{i}")
if sd_device == 'cuda' and not torch.cuda.is_available():
return
sd_steps = 1
model = ModelManager(name="sd1.5",
device=sd_device,
@ -191,15 +194,6 @@ def test_runway_sd_1_5(sd_device, strategy, sampler, cpu_textencoder, disable_ns
fx=1.3
)
assert_equal(
model,
cfg,
f"runway_sd_{strategy.capitalize()}_{name}_blur_mask.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask_blur.png",
fy=1.3
)
@pytest.mark.parametrize(
"strategy", [HDStrategy.ORIGINAL, HDStrategy.RESIZE, HDStrategy.CROP]