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
-
We shipped a Hugging Face Space that lets you calculate the memory requirements of a
DiffusionPipelinecheckpoint 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