From 7e63b389c9706b0fb0d078bf693bd5f1d74582c4 Mon Sep 17 00:00:00 2001 From: iDeNoh Date: Thu, 13 Mar 2025 13:56:36 -0500 Subject: [PATCH] Added remote vae type to infotext.py and processing_info.py --- modules/infotext.py | 2 ++ modules/processing_info.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/modules/infotext.py b/modules/infotext.py index 78c1fd92e..4ee14dd01 100644 --- a/modules/infotext.py +++ b/modules/infotext.py @@ -109,6 +109,8 @@ def parse(infotext): params[key] = False elif key == 'VAE' and val == 'TAESD': params["VAE type"] = 'Tiny' + elif key == 'VAE' and val == 'Remote': + params["VAE type"] = 'Remote' elif size is not None: params[f"{key}-1"] = int(size.group(1)) params[f"{key}-2"] = int(size.group(2)) diff --git a/modules/processing_info.py b/modules/processing_info.py index 5944855b0..fa084a2fb 100644 --- a/modules/processing_info.py +++ b/modules/processing_info.py @@ -74,6 +74,8 @@ def create_infotext(p: StableDiffusionProcessing, all_prompts=None, all_seeds=No args["VAE"] = (None if not shared.opts.add_model_name_to_info or sd_vae.loaded_vae_file is None else os.path.splitext(os.path.basename(sd_vae.loaded_vae_file))[0]) elif p.vae_type == 'Tiny': args["VAE"] = 'TAESD' + elif p.vae_type == 'Remote': + args["VAE"] = 'Remote' if shared.opts.add_model_name_to_info and getattr(shared.sd_model, 'sd_checkpoint_info', None) is not None: args["Model"] = shared.sd_model.sd_checkpoint_info.model_name.replace(',', '').replace(':', '') if shared.opts.add_model_hash_to_info and getattr(shared.sd_model, 'sd_model_hash', None) is not None: