diff options
| author | Bernard Maltais <bmaltais@gmail.com> | 2022-09-27 15:31:00 -0400 |
|---|---|---|
| committer | Bernard Maltais <bmaltais@gmail.com> | 2022-09-27 15:31:00 -0400 |
| commit | e258f89080b8ff38f040dc786290da9144860d38 (patch) | |
| tree | 80bf2e3e01073e98fd69dd977b77e8f34dc74776 | |
| parent | 20b41b7be708a501fbc7e10dceebababfd48ffda (diff) | |
Fix file name
| -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 9c3b421d..9e1efeda 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -186,7 +186,7 @@ def run_modelmerger(from_model_name, to_model_name, interp_method, interp_amount if 'model' in key and key not in theta_0:
theta_0[key] = theta_1[key]
- output_modelname = 'models/' + from_model_name + str(interp_amount) + '-' + to_model_name + str(float(1.0) - interp_amount) + '-' + interp_method.replace(" ", "_") + '-' + '-merged.ckpt'
+ output_modelname = 'models/' + from_model_name + '_' + str(interp_amount) + '-' + to_model_name + '_' + str(float(1.0) - interp_amount) + '-' + interp_method.replace(" ", "_") + '-merged.ckpt'
print(f"Saving to {output_modelname}...")
torch.save(model_0, output_modelname)
|