From 118aecfcd5f9f24f0b69a6abcafc56562f389843 Mon Sep 17 00:00:00 2001 From: maybleMyers Date: Wed, 3 Dec 2025 23:58:33 -0800 Subject: [PATCH] update model refresh --- modules_forge/main_entry.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py index 8cf41381..66bbb6aa 100644 --- a/modules_forge/main_entry.py +++ b/modules_forge/main_entry.py @@ -251,7 +251,10 @@ def refresh_model_loading_parameters(): model_data.forge_loading_parameters = dict( checkpoint_info=checkpoint_info, additional_modules=shared.opts.forge_additional_modules, - unet_storage_dtype=unet_storage_dtype + unet_storage_dtype=unet_storage_dtype, + z_transformer_dtype=getattr(shared.opts, 'z_transformer_dtype', 'Automatic'), + z_vae_dtype=getattr(shared.opts, 'z_vae_dtype', 'Automatic'), + z_text_encoder_dtype=getattr(shared.opts, 'z_text_encoder_dtype', 'Automatic'), ) print(f'Model selected: {model_data.forge_loading_parameters}')