diff --git a/comfy/sd.py b/comfy/sd.py index 97d6b2e9..67b4ff0c 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -584,7 +584,7 @@ def load_state_dict_guess_config(sd, output_vae=True, output_clip=True, output_c unet_weight_dtype.append(weight_dtype) model_config.custom_operations = model_options.get("custom_operations", None) - unet_dtype = model_options.get("weight_dtype", None) + unet_dtype = model_options.get("dtype", model_options.get("weight_dtype", None)) if unet_dtype is None: unet_dtype = model_management.unet_dtype(model_params=parameters, supported_dtypes=unet_weight_dtype)