ogkalu2 / Merge-Stable-Diffusion-models-without-distortion

Adaptation of the merging method described in the paper - Git Re-Basin: Merging Models modulo Permutation Symmetries (https://arxiv.org/abs/2209.04836) for Stable Diffusion
MIT License
139 stars 21 forks source link

Slightly faster. Changed weighted permutations to be less extreme usi… #16

Closed vaguenebula closed 1 year ago

vaguenebula commented 1 year ago

…ng sigmoid function.

List comprehension makes summation slightly faster. Also made permutations sum way faster by only adding elements that are permutated. Alpha values for permutations seemed to be extreme before, so a sigmoid function is used instead to make the alpha closer to 0.5.

Maki9009 commented 1 year ago

Hi, did you manage to get this to run on colab?