fix bug for latest model merge RAM improvement

5c94aaf2 · AUTOMATIC · 58f3ef77 · 5c94aaf2
Commit 5c94aaf2 authored Oct 17, 2022 by AUTOMATIC
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

modules/extras.py modules/extras.py +1 -0

No files found.
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -201,6 +201,7 @@ def run_modelmerger(primary_model_name, secondary_model_name, teritary_model_nam
        teritary_model = torch.load(teritary_model_info.filename, map_location='cpu')
        theta_2 = sd_models.get_state_dict_from_checkpoint(teritary_model)
    else:
+        teritary_model = None
        theta_2 = None

    theta_funcs = {