lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-05-01 03:31:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Checkpoint vae+te metadata should be in new attribute (#2073)

Browse Source 
* new metadata in new attribute
maintain compatibility with old webui by not reusing the vae metadata attribute
This commit is contained in:
DenOfEquity
2024-10-15 13:40:15 +01:00
committed by GitHub
parent 77464215c3
commit b4d2266dc6
2 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/ui_extra_networks.py
View File

@@ -214,7 +214,9 @@ class ExtraNetworksPage:
desc = metadata.get("description", None) desc = metadata.get("description", None)
if desc is not None: if desc is not None:
item["description"] = desc item["description"] = desc
vae = metadata.get("vae", None) vae = metadata.get('vae_te', None)
if vae is None: # fallback to old type
vae = metadata.get("vae", None)
if vae is not None: if vae is not None:
if isinstance(vae, str): if isinstance(vae, str):
vae = [vae] vae = [vae]

9
modules/ui_extra_networks_checkpoints_user_metadata.py
View File

@@ -17,7 +17,7 @@ class CheckpointUserMetadataEditor(ui_extra_networks_user_metadata.UserMetadataE
user_metadata = self.get_user_metadata(name) user_metadata = self.get_user_metadata(name)
user_metadata["description"] = desc user_metadata["description"] = desc
user_metadata["notes"] = notes user_metadata["notes"] = notes
user_metadata["vae"] = vae user_metadata["vae_te"] = vae
user_metadata["sd_version_str"] = 'SdVersion.' + sd_version user_metadata["sd_version_str"] = 'SdVersion.' + sd_version
self.write_user_metadata(name, user_metadata) self.write_user_metadata(name, user_metadata)
@@ -26,7 +26,12 @@ class CheckpointUserMetadataEditor(ui_extra_networks_user_metadata.UserMetadataE
user_metadata = self.get_user_metadata(name) user_metadata = self.get_user_metadata(name)
values = super().put_values_into_components(name) values = super().put_values_into_components(name)
vae = user_metadata.get('vae', None) vae = user_metadata.get('vae_te', None)
if vae is None: # fallback to old type
vae = user_metadata.get('vae', None)
if vae is not None:
if isinstance(vae, str):
vae = [vae]
version = user_metadata.get('sd_version_str', '') version = user_metadata.get('sd_version_str', '')
if version == '': if version == '':