Merge pull request #3818 from iDeNoh/dev

Added remote vae type to infotext.py and processing_info.py
This commit is contained in:
Disty0
2025-03-13 22:07:46 +03:00
committed by GitHub
2 changed files with 4 additions and 0 deletions
+2
View File
@@ -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))
+2
View File
@@ -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: