From 6a35eefdd54f370553b0988f981bfe960f507f80 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 7 May 2025 17:52:47 +0300 Subject: [PATCH] Update model_optimization_nodes.py --- nodes/model_optimization_nodes.py | 1 + 1 file changed, 1 insertion(+) diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py index 93d4caf..127c7f9 100644 --- a/nodes/model_optimization_nodes.py +++ b/nodes/model_optimization_nodes.py @@ -226,6 +226,7 @@ class CheckpointLoaderKJ(BaseLoaderKJ): diffusion_model_prefix = model_detection.unet_prefix_from_state_dict(sd) parameters = comfy.utils.calculate_parameters(sd, diffusion_model_prefix) + weight_dtype = comfy.utils.weight_dtype(sd, diffusion_model_prefix) load_device = mm.get_torch_device() model_config = model_detection.model_config_from_unet(sd, diffusion_model_prefix, metadata=metadata)