diff --git a/modules/extras.py b/modules/extras.py index 6f05b0fe..306c5609 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -150,14 +150,14 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_ if theta_func2: shared.state.textinfo = "Loading B" print(f"Loading {secondary_model_info.filename}...") - theta_1 = sd_models.read_state_dict(secondary_model_info.filename, map_location='cpu') + theta_1 = sd_models.load_torch_file(secondary_model_info.filename) else: theta_1 = None if theta_func1: shared.state.textinfo = "Loading C" print(f"Loading {tertiary_model_info.filename}...") - theta_2 = sd_models.read_state_dict(tertiary_model_info.filename, map_location='cpu') + theta_2 = sd_models.load_torch_file(tertiary_model_info.filename) shared.state.textinfo = 'Merging B and C' shared.state.sampling_steps = len(theta_1.keys())