Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
83b112a1
"vscode:/vscode.git/clone" did not exist on "0e5aee46e0fcedbb76a1955ab2663e9c494f3b91"
Unverified
Commit
83b112a1
authored
Jul 01, 2024
by
Sayak Paul
Committed by
GitHub
Jul 01, 2024
Browse files
shift cache in benchmarking. (#8740)
* shift cache. * comment
parent
8690e8b9
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
6 deletions
+4
-6
.github/workflows/benchmark.yml
.github/workflows/benchmark.yml
+1
-1
benchmarks/run_all.py
benchmarks/run_all.py
+3
-5
No files found.
.github/workflows/benchmark.yml
View file @
83b112a1
...
@@ -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
...
...
benchmarks/run_all.py
View file @
83b112a1
...
@@ -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"
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment