Commit f894dd55 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix for broken checkpoint merger

parent 9931c0bd

modules/sd_models.py

+4 −1
Original line number Diff line number Diff line
@@ -148,7 +148,10 @@ def get_state_dict_from_checkpoint(pl_sd): 
        if new_key is not None:

            sd[new_key] = v



    return sd

    pl_sd.clear()

    pl_sd.update(sd)



    return pl_sd





def load_model_weights(model, checkpoint_info):

CRA Git | Maintained and supported by SUSTech CRA and CCSE