model_options consistency between functions.

weight_dtype -> dtype
comfyanonymous · Oct 12, 2024 · 6632365 · 6632365
1 parent ad07796
commit 6632365
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)