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,
device=torch.device(device),
disable_nsfw=True,
sd_cpu_textencoder=True,
sd_cpu_textencoder=device == "cuda",
enable_controlnet=True,
controlnet_method=controlnet_method,
)
@ -111,7 +111,7 @@ def test_local_file_path(device, local_file):
assert_equal(
model,
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",
mask_p=current_dir / "overture-creations-5sI6fQgYIuo_mask.png",
)