• Lincoln Stein's avatar
    [Feat] checkpoint_merger works on local models as well as ones that use safetensors (#2060) · 7533e3d7
    Lincoln Stein authored
    
    
    * 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: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Update examples/community/checkpoint_merger.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * Update examples/community/checkpoint_merger.py
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    
    * fix import style error
    Co-authored-by: default avatarLincoln Stein <lstein@gmail.com>
    Co-authored-by: default avatarPatrick von Platen <patrick.v.platen@gmail.com>
    7533e3d7
checkpoint_merger.py 12.5 KB