-
Notifications
You must be signed in to change notification settings - Fork 62
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix diffusion caching #594
Conversation
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
fixes huggingface#593 Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
999bec6
to
c988fb7
Compare
@@ -698,6 +699,7 @@ def _export( | |||
save_dir = TemporaryDirectory() | |||
save_dir_path = Path(save_dir.name) | |||
# 1. Fetch all model configs | |||
input_shapes_copy = copy.deepcopy(input_shapes) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nit: maybe just shadow input_shapes
and do:
input_shapes_copy = copy.deepcopy(input_shapes) | |
input_shapes = copy.deepcopy(input_shapes) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
this will shadow it but before it's actually updated in the call below, so the problem won't be fixed, no ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ok, no worries then, let's keep it like that.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Gently pinging @dacorvo since INF2 test do not pass. It seems to be unrelated but double checking.
I don't know how the CI was triggered, but it does not seem to have the correct token, which explains the error. |
fixes #593