"vscode:/vscode.git/clone" did not exist on "0e5aee46e0fcedbb76a1955ab2663e9c494f3b91"
Unverified Commit 83b112a1 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

shift cache in benchmarking. (#8740)

* shift cache.

* comment
parent 8690e8b9
...@@ -22,7 +22,7 @@ jobs: ...@@ -22,7 +22,7 @@ jobs:
runs-on: [single-gpu, nvidia-gpu, a10, ci] runs-on: [single-gpu, nvidia-gpu, a10, ci]
container: container:
image: diffusers/diffusers-pytorch-compile-cuda 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: steps:
- name: Checkout diffusers - name: Checkout diffusers
uses: actions/checkout@v3 uses: actions/checkout@v3
......
...@@ -39,11 +39,8 @@ def main(): ...@@ -39,11 +39,8 @@ def main():
for file in python_files: for file in python_files:
print(f"****** Running file: {file} ******") print(f"****** Running file: {file} ******")
if "ip_adapters" in file:
continue
# Run with canonical settings. # 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}" command = f"python {file}"
run_command(command.split()) run_command(command.split())
...@@ -52,7 +49,8 @@ def main(): ...@@ -52,7 +49,8 @@ def main():
# Run variants. # Run variants.
for file in python_files: 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 continue
if file == "benchmark_text_to_image.py": if file == "benchmark_text_to_image.py":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment