-
Notifications
You must be signed in to change notification settings - Fork 6.3k
Deprecate slicing and tiling methods from DiffusionPipeline
#12271
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
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. |
@@ -1705,6 +1705,12 @@ def enable_vae_tiling(self): | |||
compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow | |||
processing larger images. | |||
""" | |||
depr_message = "Calling `enable_vae_tiling()` on a `DiffusionPipeline` is deprecated and this method will be removed in a future version. Please use `pipe.vae.enable_tiling()`." |
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.
Perhaps use "Calling enable_vae_tiling()
on a {cls.__name__}
is deprecated`? So the specific pipeline name shows up in the message?
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.
Looks good. Just minor comment about the warning message.
What does this PR do?
Calling
pipe.enable_vae_tiling()
is same callingpipe.vae.enable_tiling()
. Same applies for slicing, too.We can remove these methods to encourage users to be more explicit (in this case even the line of code doesn't change).
This PR starts a deprecation cycle for these methods.