Skip to content

Commit 6632365

Browse files
model_options consistency between functions.
weight_dtype -> dtype
1 parent ad07796 commit 6632365

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

comfy/sd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -584,7 +584,7 @@ def load_state_dict_guess_config(sd, output_vae=True, output_clip=True, output_c
584584
unet_weight_dtype.append(weight_dtype)
585585

586586
model_config.custom_operations = model_options.get("custom_operations", None)
587-
unet_dtype = model_options.get("weight_dtype", None)
587+
unet_dtype = model_options.get("dtype", model_options.get("weight_dtype", None))
588588

589589
if unet_dtype is None:
590590
unet_dtype = model_management.unet_dtype(model_params=parameters, supported_dtypes=unet_weight_dtype)

0 commit comments

Comments
 (0)