Mirrors/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2025-12-28 05:35:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update model refresh

Browse Source
This commit is contained in:
maybleMyers 2025-12-03 23:58:33 -08:00
parent f45647b101
commit 118aecfcd5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules_forge/main_entry.py
View File

@ -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}')