Muhammad Rizqi Nur R-N
R-N
/ replace_vae.py
Last active
February 23, 2023 17:31
https://gist.github.com/ProGamerGov/70061a08e3a2da6e9ed83e145ea24a70 https://github.com/CrazyBoyM/merge-models/blob/main/merge.py
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # Script by https://github.com/ProGamerGov | |
| import copy | |
| import torch | |
| import argparse | |
| from safetensors.torch import load_file, save_file | |
| parser = argparse.ArgumentParser(description="Merge model and vae") | |
| parser.add_argument("model", type=str, help="Path to model") | |
| parser.add_argument("vae", type=str, help="Path to vae") |