diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-27 20:55:54 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-09-27 20:55:54 +0300 |
| commit | a902cbff4faf04b13d22ddaeb4bbe643fe7e8276 (patch) | |
| tree | 7a40fad9ae2f888d46987413a010ec08615c00d1 /modules | |
| parent | ca358c14512a7d31846f9aea3a1c46e4a8e972d8 (diff) | |
| parent | 0bd8aa937f9172b44de30d158258c51b2174f136 (diff) | |
Merge pull request #1166 from bmaltais/model-merge-upd
Add interpolation method and weight to merged model output filename
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/extras.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/extras.py b/modules/extras.py index 15873204..c4ee2b62 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -186,7 +186,7 @@ def run_modelmerger(modelname_0, modelname_1, interp_method, interp_amount): if 'model' in key and key not in theta_0:
theta_0[key] = theta_1[key]
- output_modelname = 'models/' + modelname_0 + '-' + modelname_1 + '-merged.ckpt'
+ output_modelname = 'models/' + modelname_0 + '-' + modelname_1 + '-' + interp_method.replace(" ", "_") + '-' + str(interp_amount) + '-merged.ckpt'
print(f"Saving to {output_modelname}...")
torch.save(model_0, output_modelname)
|