From f10359989f0ae0462041e99013d002766e0ec74d Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Sat, 10 Aug 2024 19:40:56 -0700 Subject: [PATCH] fix --- modules_forge/main_entry.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules_forge/main_entry.py b/modules_forge/main_entry.py index 1becebbd..5eddbc47 100644 --- a/modules_forge/main_entry.py +++ b/modules_forge/main_entry.py @@ -126,7 +126,7 @@ def refresh_model_loading_parameters(): model_data.forge_loading_parameters = dict( checkpoint_info=checkpoint_info, vae_filename=vae_resolution.vae, - unet_storage_dtype=forge_unet_storage_dtype_options[shared.opts.forge_unet_storage_dtype] + unet_storage_dtype=forge_unet_storage_dtype_options.get(shared.opts.forge_unet_storage_dtype, None) ) print(f'Model selected: {model_data.forge_loading_parameters}')