From 98aa6a9c8786e7202d3014b19420cf86eb223847 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sun, 30 Jun 2024 09:27:31 +0530 Subject: [PATCH 1/2] shift cache. --- .github/workflows/benchmark.yml | 2 +- benchmarks/run_all.py | 7 ++----- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml index 00c9cbed0636..d638d1435dfc 100644 --- a/.github/workflows/benchmark.yml +++ b/.github/workflows/benchmark.yml @@ -22,7 +22,7 @@ jobs: runs-on: [single-gpu, nvidia-gpu, a10, ci] container: image: diffusers/diffusers-pytorch-compile-cuda - options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface/diffusers:/mnt/cache/ --gpus 0 + options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0 steps: - name: Checkout diffusers uses: actions/checkout@v3 diff --git a/benchmarks/run_all.py b/benchmarks/run_all.py index 96d30c0837c8..e43fe532a289 100644 --- a/benchmarks/run_all.py +++ b/benchmarks/run_all.py @@ -39,11 +39,8 @@ def main(): for file in python_files: print(f"****** Running file: {file} ******") - if "ip_adapters" in file: - continue - # Run with canonical settings. - if file != "benchmark_text_to_image.py": + if file != "benchmark_text_to_image.py" and file != "benchmark_ip_adapters.py": command = f"python {file}" run_command(command.split()) @@ -52,7 +49,7 @@ def main(): # Run variants. for file in python_files: - if "ip_adapters" in file: + if file == "benchmark_ip_adapters.py": continue if file == "benchmark_text_to_image.py": From 543d5659a259245291c01c37dbecba353d577395 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sun, 30 Jun 2024 09:48:13 +0530 Subject: [PATCH 2/2] comment --- benchmarks/run_all.py | 1 + 1 file changed, 1 insertion(+) diff --git a/benchmarks/run_all.py b/benchmarks/run_all.py index e43fe532a289..c9932cc71c38 100644 --- a/benchmarks/run_all.py +++ b/benchmarks/run_all.py @@ -49,6 +49,7 @@ def main(): # Run variants. for file in python_files: + # See: https://github.com/pytorch/pytorch/issues/129637 if file == "benchmark_ip_adapters.py": continue