Won-Kyu Park wkpark
wkpark
/ replace_vae.py
Last active
June 22, 2023 16:08
— forked from ProGamerGov/replace_vae.py
Replace the VAE in a Stable Diffusion model with a new VAE. Tested on v1.4 & v1.5 SD models
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 | |
| # | |
| # ChangeLog: | |
| # - support safetensors, save float16 if needed, check filename 2023/06/22 by wkpark | |
| # | |
| import os | |
| import sys | |
| import copy | |
| import torch |