VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

change ver no for Diffusers VAE changing

Browse Source
This commit is contained in:
Kohya S
2023-06-12 12:29:44 +09:00
parent 0dfffcd88a
commit 67f09b7d7e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/model_util.py
View File

@@ -128,7 +128,7 @@ def renew_vae_attention_paths(old_list, n_shave_prefix_segments=0):
new_item = new_item.replace("norm.weight", "group_norm.weight") new_item = new_item.replace("norm.weight", "group_norm.weight")
new_item = new_item.replace("norm.bias", "group_norm.bias") new_item = new_item.replace("norm.bias", "group_norm.bias")
if diffusers.__version__ < "0.15.0": if diffusers.__version__ < "0.17.0":
new_item = new_item.replace("q.weight", "query.weight") new_item = new_item.replace("q.weight", "query.weight")
new_item = new_item.replace("q.bias", "query.bias") new_item = new_item.replace("q.bias", "query.bias")
@@ -207,7 +207,7 @@ def assign_to_checkpoint(
# proj_attn.weight has to be converted from conv 1D to linear # proj_attn.weight has to be converted from conv 1D to linear
reshaping = False reshaping = False
if diffusers.__version__ < "0.15.0": if diffusers.__version__ < "0.17.0":
if "proj_attn.weight" in new_path: if "proj_attn.weight" in new_path:
reshaping = True reshaping = True
else: else:
@@ -802,7 +802,7 @@ def convert_vae_state_dict(vae_state_dict):
sd_mid_res_prefix = f"mid.block_{i+1}." sd_mid_res_prefix = f"mid.block_{i+1}."
vae_conversion_map.append((sd_mid_res_prefix, hf_mid_res_prefix)) vae_conversion_map.append((sd_mid_res_prefix, hf_mid_res_prefix))
if diffusers.__version__ < "0.15.0": if diffusers.__version__ < "0.17.0":
vae_conversion_map_attn = [ vae_conversion_map_attn = [
# (stable-diffusion, HF Diffusers) # (stable-diffusion, HF Diffusers)
("norm.", "group_norm."), ("norm.", "group_norm."),