Skip to content

Commit

Permalink
shift cache in benchmarking. (#8740)
Browse files Browse the repository at this point in the history
* shift cache.

* comment
  • Loading branch information
sayakpaul committed Dec 23, 2024
1 parent 5f10c18 commit f9ba2ff
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/benchmark.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
8 changes: 3 additions & 5 deletions benchmarks/run_all.py
Original file line number Diff line number Diff line change
Expand Up @@ -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())

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

# Run variants.
for file in python_files:
if "ip_adapters" in file:
# See: https://github.com/pytorch/pytorch/issues/129637
if file == "benchmark_ip_adapters.py":
continue

if file == "benchmark_text_to_image.py":
Expand Down

0 comments on commit f9ba2ff

Please sign in to comment.