diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-17 08:28:18 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-17 08:28:18 +0300 |
| commit | 5c94aaf290f8ad7bf4499a91c268ad0791b0432f (patch) | |
| tree | 4e6f9072c7ea84fa266258b52cb6fd5e3c8d0346 /modules/extras.py | |
| parent | 58f3ef77336663bce2321f5b692cf2aeacd3ac1c (diff) | |
fix bug for latest model merge RAM improvement
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index 340a45fd..8dbab240 100644 --- 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 = {
|