Commit 7c016dd6 authored by space-nuko's avatar space-nuko
Browse files

Calculate shorthash on merge if not exist

parent afc349c2

modules/extras.py

+1 −0
Original line number Diff line number Diff line
@@ -263,6 +263,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_ 
        metadata["merge_recipe"] = json.dumps(merge_recipe)



        def add_model_metadata(checkpoint_info):

            checkpoint_info.calculate_shorthash()

            metadata["models"][checkpoint_info.sha256] = {

                "name": checkpoint_info.name,

                "legacy_hash": checkpoint_info.hash,

CRA Git | Maintained and supported by SUSTech CRA and CCSE