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

fix test

Browse Source
This commit is contained in:
Qing 2023-02-01 21:36:37 +08:00
parent 5950cc9c38
commit 33a1bde647
2 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
lama_cleaner/tests/overture-creations-5sI6fQgYIuo_all_mask.png Normal file
View File

26
lama_cleaner/tests/test_instruct_pix2pix.py
View File

@ -24,7 +24,7 @@ def test_instruct_pix2pix(disable_nsfw, cpu_offload):
disable_nsfw=disable_nsfw, disable_nsfw=disable_nsfw,
sd_cpu_textencoder=False, sd_cpu_textencoder=False,
cpu_offload=cpu_offload) cpu_offload=cpu_offload)
cfg = get_config(strategy=HDStrategy.ORIGINAL, prompt='make it good', p2p_steps=sd_steps, sd_scale=1.1) cfg = get_config(strategy=HDStrategy.ORIGINAL, prompt='What if it were snowing?', p2p_steps=sd_steps, sd_scale=1.1)
name = f"device_{device}_disnsfw_{disable_nsfw}_cpu_offload_{cpu_offload}" name = f"device_{device}_disnsfw_{disable_nsfw}_cpu_offload_{cpu_offload}"
@ -36,3 +36,27 @@ def test_instruct_pix2pix(disable_nsfw, cpu_offload):
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
fx=1.3 fx=1.3
) )
@pytest.mark.parametrize("disable_nsfw", [False])
@pytest.mark.parametrize("cpu_offload", [False])
def test_instruct_pix2pix_snow(disable_nsfw, cpu_offload):
sd_steps = 50 if device == 'cuda' else 1
model = ModelManager(name="instruct_pix2pix",
device=torch.device(device),
hf_access_token="",
sd_run_local=True,
disable_nsfw=disable_nsfw,
sd_cpu_textencoder=False,
cpu_offload=cpu_offload)
cfg = get_config(strategy=HDStrategy.ORIGINAL, prompt='What if it were snowing?', p2p_steps=sd_steps)
name = f"snow"
assert_equal(
model,
cfg,
f"instruct_pix2pix_{name}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)