Calculate the component-wise memory of DiffusionPipeline
checkpoint
#7434
sayakpaul
started this conversation in
Show and tell
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
Uh oh!
There was an error while loading. Please reload this page.
-
We shipped a Hugging Face Space that lets you calculate the memory requirements of a
DiffusionPipeline
checkpoint given atorch_dtype
:https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading#diffusion-pipeline
Scroll down the section a bit and you should be able to find out a para saying "You can use the Space below to gauge the memory requirements ...".
It doesn't require downloading the pipeline checkpoints.
Beta Was this translation helpful? Give feedback.
All reactions