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
torch.load can be unsafe due to pickle exploits #32
torch.load can be unsafe due to pickle exploits, most projects use a safe loader of some description if they support torch checkpoints.
You could support safetensors also to mitigate that. The code for safe tensors is pretty simple: