diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-28 13:19:47 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-09-28 13:19:47 +0000 |
| commit | e22ea454a273b3a8a807a5acb2e6f0d0d41c9aa7 (patch) | |
| tree | 0c131db0e2f3eba171612646a4cf0ffd4a45de08 /modules/extras.py | |
| parent | 5eb9d1aeac8dcfb2073f57821c765a339ccde7f6 (diff) | |
| parent | 228a2f30e7f8ae1a752c68ac189e9ca6bb4e29f6 (diff) | |
| download | stable-diffusion-webui-gfx803-e22ea454a273b3a8a807a5acb2e6f0d0d41c9aa7.tar.gz stable-diffusion-webui-gfx803-e22ea454a273b3a8a807a5acb2e6f0d0d41c9aa7.tar.bz2 stable-diffusion-webui-gfx803-e22ea454a273b3a8a807a5acb2e6f0d0d41c9aa7.zip | |
Merge pull request #1222 from bmaltais/inverse-sigmoid-interpolation
(feat): Checkpoint merge using Inverse sigmoid interpolation
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index b8ebc619..15de033a 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -150,6 +150,12 @@ def run_modelmerger(primary_model_name, secondary_model_name, interp_method, int alpha = alpha * alpha * (3 - (2 * alpha))
return theta0 + ((theta1 - theta0) * alpha)
+ # Inverse Smoothstep (https://en.wikipedia.org/wiki/Smoothstep)
+ def inv_sigmoid(theta0, theta1, alpha):
+ import math
+ alpha = 0.5 - math.sin(math.asin(1.0 - 2.0 * alpha) / 3.0)
+ return theta0 + ((theta1 - theta0) * alpha)
+
if os.path.exists(primary_model_name):
primary_model_filename = primary_model_name
primary_model_name = os.path.splitext(os.path.basename(primary_model_name))[0]
@@ -174,6 +180,7 @@ def run_modelmerger(primary_model_name, secondary_model_name, interp_method, int theta_funcs = {
"Weighted Sum": weighted_sum,
"Sigmoid": sigmoid,
+ "Inverse Sigmoid": inv_sigmoid
}
theta_func = theta_funcs[interp_method]
|