>>102001089
sd_models.read_state_dict() has been removed (pass) but its still called during merging
diff --git a/modules/extras.py b/modules/extras.py
index 2a310ae3..08fbc5b4 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -179,7 +179,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
shared.state.textinfo = f"Loading {primary_model_info.filename}..."
print(f"Loading {primary_model_info.filename}...")
- theta_0 = sd_models.read_state_dict(primary_model_info.filename, map_location='cpu')
+ theta_0 = sd_models.load_torch_file(primary_model_info.filename)^M
print("Merging...")
shared.state.textinfo = 'Merging A and B'
@@ -222,7 +222,7 @@ def run_modelmerger(id_task, primary_model_name, secondary_model_name, tertiary_
if bake_in_vae_filename is not None:
print(f"Baking in VAE from {bake_in_vae_filename}")
shared.state.textinfo = 'Baking in VAE'
- vae_dict = sd_vae.load_vae_dict(bake_in_vae_filename, map_location='cpu')
+ vae_dict = sd_vae.load_torch_file(bake_in_vae_filename)^M
for key in vae_dict.keys():
theta_0_key = 'first_stage_model.' + key