From 7545d54870e4a661d9f9926348ab8d814bd04775 Mon Sep 17 00:00:00 2001 From: Qing Date: Wed, 10 Jan 2024 21:24:59 +0800 Subject: [PATCH] fix test --- iopaint/tests/test_controlnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/iopaint/tests/test_controlnet.py b/iopaint/tests/test_controlnet.py index 4f5f800..30c926a 100644 --- a/iopaint/tests/test_controlnet.py +++ b/iopaint/tests/test_controlnet.py @@ -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", )