saftle
saftle
/ replace_vae.py
Last active
January 21, 2023 10:00
— forked from ProGamerGov/replace_vae.py
Fixed script to work with models that were merged with Automatic1111, and included the numpy dependency which the script uses.
This file contains 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 numpy as np | |
| # Path to model and VAE files that you want to merge | |
| vae_file_path = "vae-ft-mse-840000-ema-pruned.ckpt" | |
| model_file_path = "v1-5-pruned-emaonly.ckpt" | |
| # Name to use for new model file |