From c8f6eefd54ceae8abb69abfecaa362ef843513ff Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 5 Dec 2023 15:34:09 +0530 Subject: [PATCH] fix: condition --- benchmarks/run_all.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/run_all.py b/benchmarks/run_all.py index 5753db2d72a8..bef93b4ab31a 100644 --- a/benchmarks/run_all.py +++ b/benchmarks/run_all.py @@ -58,7 +58,7 @@ def main(): elif file in ["benchmark_controlnet.py", "benchmark_t2i_adapter.py"]: sdxl_ckpt = ( "diffusers/controlnet-canny-sdxl-1.0" - if "controlnet" == file + if "controlnet" in file else "TencentARC/t2i-adapter-canny-sdxl-1.0" ) command = f"python {file} --ckpt {sdxl_ckpt}"