From 04af2c8d63744fb6c02d35d328a2c84380cca444 Mon Sep 17 00:00:00 2001 From: s1dlx Date: Sun, 21 Jul 2024 22:34:26 +0100 Subject: [PATCH] fmt --- sd_meh/rebasin.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/sd_meh/rebasin.py b/sd_meh/rebasin.py index 2fbb418..010d67f 100644 --- a/sd_meh/rebasin.py +++ b/sd_meh/rebasin.py @@ -2200,11 +2200,9 @@ def apply_permutation(ps: PermutationSpec, perm, params): def update_model_a(ps: PermutationSpec, perm, model_a, new_alpha): for k in model_a: try: - perm_params = get_permuted_param( - ps, perm, k, model_a - ) + perm_params = get_permuted_param(ps, perm, k, model_a) model_a[k] = model_a[k] * (1 - new_alpha) + new_alpha * perm_params - except RuntimeError: # dealing with pix2pix and inpainting models + except RuntimeError: # dealing with pix2pix and inpainting models continue return model_a