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

fix test

Browse Source
This commit is contained in:
Qing 2024-01-10 21:24:59 +08:00
parent cbcdf3b9a2
commit 7545d54870
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
iopaint/tests/test_controlnet.py
View File

@ -29,7 +29,7 @@ def test_runway_sd_1_5(device, controlnet_method):
name=model_name, name=model_name,
device=torch.device(device), device=torch.device(device),
disable_nsfw=True, disable_nsfw=True,
sd_cpu_textencoder=True, sd_cpu_textencoder=device == "cuda",
enable_controlnet=True, enable_controlnet=True,
controlnet_method=controlnet_method, controlnet_method=controlnet_method,
) )
@ -111,7 +111,7 @@ def test_local_file_path(device, local_file):
assert_equal( assert_equal(
model, model,
cfg, cfg,
f"{controlnet_kwargs['controlnet_method']}_local_model_{name}.png", f"{convert_controlnet_method_name(controlnet_kwargs['controlnet_method'])}_local_model_{name}.png",
img_p=current_dir / "overture-creations-5sI6fQgYIuo.png", img_p=current_dir / "overture-creations-5sI6fQgYIuo.png",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png", mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
) )