Skip to content

Commit 83b112a

Browse files
authored
shift cache in benchmarking. (#8740)
* shift cache. * comment
1 parent 8690e8b commit 83b112a

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

.github/workflows/benchmark.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ jobs:
2222
runs-on: [single-gpu, nvidia-gpu, a10, ci]
2323
container:
2424
image: diffusers/diffusers-pytorch-compile-cuda
25-
options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface/diffusers:/mnt/cache/ --gpus 0
25+
options: --shm-size "16gb" --ipc host -v /mnt/hf_cache:/mnt/cache/ --gpus 0
2626
steps:
2727
- name: Checkout diffusers
2828
uses: actions/checkout@v3

benchmarks/run_all.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,8 @@ def main():
3939
for file in python_files:
4040
print(f"****** Running file: {file} ******")
4141

42-
if "ip_adapters" in file:
43-
continue
44-
4542
# Run with canonical settings.
46-
if file != "benchmark_text_to_image.py":
43+
if file != "benchmark_text_to_image.py" and file != "benchmark_ip_adapters.py":
4744
command = f"python {file}"
4845
run_command(command.split())
4946

@@ -52,7 +49,8 @@ def main():
5249

5350
# Run variants.
5451
for file in python_files:
55-
if "ip_adapters" in file:
52+
# See: https://github.com/pytorch/pytorch/issues/129637
53+
if file == "benchmark_ip_adapters.py":
5654
continue
5755

5856
if file == "benchmark_text_to_image.py":

0 commit comments

Comments
 (0)