Adding some safetensors docs.
#2122
Conversation
|
The documentation is not available anymore as the PR was closed or merged. |
| @@ -0,0 +1,19 @@ | |||
| # What is safetensors ? | |||
There was a problem hiding this comment.
I think we can delete this file no? :-)
| ``` | ||
| from diffusers import StableDiffusionPipeline | ||
|
|
||
| pipe = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", use_safetensors=True) |
There was a problem hiding this comment.
We don't have this argument yet and I think it might take 1,2 weeks to align with transfomers. Maybe just do the hacky way via:
pip install safetensors
pip uninstall safetensors
? for now?
There was a problem hiding this comment.
I can revisit what I've written to conform what is actually here now.
There was a problem hiding this comment.
Super nice doc :-) I'll try to focus on getting a solution for #2123 asap
I removed the problematic part of the doc, so we can merge this now, and readd whatever is missing from this document when we have a solution to force load specific weights |
|
Hello, actually it's not clear from the doc how to load *.safetensors using |
* Tmp. * Adding more docs. * Doc style. * Remove the argument `use_safetensors=True`. * doc-builder
Written with #2123 in mind.
This is optional but makes the actual inner working more explicit for users.