From b4d2266dc61b77059de9047f27854e79946d21e5 Mon Sep 17 00:00:00 2001
From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com>
Date: Tue, 15 Oct 2024 13:40:15 +0100
Subject: [PATCH] Checkpoint vae+te metadata should be in new attribute (#2073)

* new metadata in new attribute
maintain compatibility with old webui by not reusing the vae metadata attribute
---
 modules/ui_extra_networks.py                           | 4 +++-
 modules/ui_extra_networks_checkpoints_user_metadata.py | 9 +++++++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/modules/ui_extra_networks.py b/modules/ui_extra_networks.py
index 3d2baa00..cd3ec346 100644
--- a/modules/ui_extra_networks.py
+++ b/modules/ui_extra_networks.py
@@ -214,7 +214,9 @@ class ExtraNetworksPage:
         desc = metadata.get("description", None)
         if desc is not None:
             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 isinstance(vae, str):
                 vae = [vae]
diff --git a/modules/ui_extra_networks_checkpoints_user_metadata.py b/modules/ui_extra_networks_checkpoints_user_metadata.py
index b7c66382..fceb7c8f 100644
--- a/modules/ui_extra_networks_checkpoints_user_metadata.py
+++ b/modules/ui_extra_networks_checkpoints_user_metadata.py
@@ -17,7 +17,7 @@ class CheckpointUserMetadataEditor(ui_extra_networks_user_metadata.UserMetadataE
         user_metadata = self.get_user_metadata(name)
         user_metadata["description"] = desc
         user_metadata["notes"] = notes
-        user_metadata["vae"] = vae
+        user_metadata["vae_te"] = vae
         user_metadata["sd_version_str"] = 'SdVersion.' + sd_version
 
         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)
         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', '')
         if version == '':