File tree 2 files changed +4
-6
lines changed
2 files changed +4
-6
lines changed Original file line number Diff line number Diff line change 22
22
runs-on : [single-gpu, nvidia-gpu, a10, ci]
23
23
container :
24
24
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
26
26
steps :
27
27
- name : Checkout diffusers
28
28
uses : actions/checkout@v3
Original file line number Diff line number Diff line change @@ -39,11 +39,8 @@ def main():
39
39
for file in python_files :
40
40
print (f"****** Running file: { file } ******" )
41
41
42
- if "ip_adapters" in file :
43
- continue
44
-
45
42
# 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" :
47
44
command = f"python { file } "
48
45
run_command (command .split ())
49
46
@@ -52,7 +49,8 @@ def main():
52
49
53
50
# Run variants.
54
51
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" :
56
54
continue
57
55
58
56
if file == "benchmark_text_to_image.py" :
You can’t perform that action at this time.
0 commit comments