-
Notifications
You must be signed in to change notification settings - Fork 6.1k
[Stable Diffusion] Allow users to disable Safety checker if loading model from checkpoint #2768
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
…ls from .ckpt or .safetensors
The documentation is not available anymore as the PR was closed or merged. |
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.
Well done!
@patrickvonplaten could you check here? |
I'm surprised this was not a thing until now. Does that mean saving a model as fp16 was not possible until now? Hope this gets merged asap. |
I think this is a duplicate of: #2733 no? |
@patrickvonplaten I will try to check when I get home, thanks for pointing that out |
I had a look at the stuff you sent and you were correct, thanks for pointing that out - I wasn't aware that this was possible. |
Failed test is not caused by this PR as #2817 is facing the same issue |
Thanks for adapting the PR - think we can merge this one |
…odel from checkpoint (huggingface#2768) * Allow user to disable SafetyChecker and enable dtypes if loading models from .ckpt or .safetensors * Fix Import sorting (Ruff error) * Get rid of the dtype convert method as it was implemented all along * Fix the docstring * Fix ruff formatting --------- Co-authored-by: Patrick von Platen <[email protected]>
…odel from checkpoint (huggingface#2768) * Allow user to disable SafetyChecker and enable dtypes if loading models from .ckpt or .safetensors * Fix Import sorting (Ruff error) * Get rid of the dtype convert method as it was implemented all along * Fix the docstring * Fix ruff formatting --------- Co-authored-by: Patrick von Platen <[email protected]>
…odel from checkpoint (huggingface#2768) * Allow user to disable SafetyChecker and enable dtypes if loading models from .ckpt or .safetensors * Fix Import sorting (Ruff error) * Get rid of the dtype convert method as it was implemented all along * Fix the docstring * Fix ruff formatting --------- Co-authored-by: Patrick von Platen <[email protected]>
Benefits: The user will have the option to save both VRAM (safety checker and float16) and disk space (safety checker download)
Intended use:
Closes #2755