diffusers
7533e3d7 - [Feat] checkpoint_merger works on local models as well as ones that use safetensors (#2060)

Commit
2 years ago
[Feat] checkpoint_merger works on local models as well as ones that use safetensors (#2060) * allow a local model directory to be used for merging * moved checkpoint merge bugfix into main for testing * possibly fix local variable "config_dict" referenced before assignment * fix deprecation warning * debugging... * debugging * allow safetensors * safetensors try again * fix syntax error * further debugging * fix logic error when checkpoint 2 is none * more debugging... * more debuging... * more debugging... * more debugging... * debugging * clean up status reporting * skip the requires_safety_checker boolean * moved checkpoint merge bugfix into main for testing * possibly fix local variable "config_dict" referenced before assignment * fix deprecation warning * allow safetensors * fix logic error when checkpoint 2 is none * clean up status reporting * undo hack to use private repo for community pipelines * allow a local model directory to be used for merging * allow safetensors * clean up status reporting * reformatted with black * sort imported modules correctly * Update examples/community/checkpoint_merger.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update examples/community/checkpoint_merger.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Update examples/community/checkpoint_merger.py Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * fix import style error Co-authored-by: Lincoln Stein <lstein@gmail.com> Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Author
Parents
Loading