sdxl for other benchmarks.

huggingface · Dec 5, 2023 · 9a9d5ea · 9a9d5ea
1 parent 63aee79
commit 9a9d5ea
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/benchmarks/run_all.py b/benchmarks/run_all.py
@@ -47,6 +47,26 @@ def main():
                 command += " --run_compile"
                 run_command(command.split())
 
+        elif file in ["benchmark_sd_img.py", "benchmark_sd_inpainting.py"]:
+            sdxl_ckpt = "stabilityai/stable-diffusion-xl-base-1.0"
+            command = f"python {file} --ckpt {sdxl_ckpt}"
+            run_command(command.split())
+
+            command += " --run_compile"
+            run_command(command.split())
+
+        elif file in ["benchmark_controlnet.py", "benchmark_t2i_adapter.py"]:
+            sdxl_ckpt = (
+                "diffusers/controlnet-canny-sdxl-1.0"
+                if "controlnet" == file
+                else "TencentARC/t2i-adapter-canny-sdxl-1.0"
+            )
+            command = f"python {file} --ckpt {sdxl_ckpt}"
+            run_command(command.split())
+
+            command += " --run_compile"
+            run_command(command.split())
+
 
 if __name__ == "__main__":
     main()