StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Calculate shorthash on merge if not exist

Browse Source
This commit is contained in:
space-nuko 2023-04-02 19:06:39 -05:00
parent afc349c2c0
commit 7c016dd642
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
modules/extras.py
View File

@ -263,6 +263,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
metadata["merge_recipe"] = json.dumps(merge_recipe) metadata["merge_recipe"] = json.dumps(merge_recipe)
def add_model_metadata(checkpoint_info): def add_model_metadata(checkpoint_info):
checkpoint_info.calculate_shorthash()
metadata["models"][checkpoint_info.sha256] = { metadata["models"][checkpoint_info.sha256] = {
"name": checkpoint_info.name, "name": checkpoint_info.name,
"legacy_hash": checkpoint_info.hash, "legacy_hash": checkpoint_info.hash,