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

Feature request (Dreaming): Algorithm 3 "MERGEMANY" #47

Open 6DammK9 opened 4 months ago

6DammK9 commented 4 months ago

From the paper, I see Algorithm 1 and 2 are implemented. Using scipy for the SOLVELAP look nice to me. However in the bottom session, there is one more algorithm called "MERGEMANY", it iterlates for each model in a OVR manner, for a large model pool (paper is around 10). Looks like it requires a new codebase to archieve this? I think TIES is a lot more simple in this situation

image