From e5ff6a1b53211ce3130cc0de071ce137714e03a4 Mon Sep 17 00:00:00 2001 From: strint Date: Thu, 16 Oct 2025 22:47:03 +0800 Subject: [PATCH] refine log --- comfy/model_base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/comfy/model_base.py b/comfy/model_base.py index 75d469221..b0bb0cfb0 100644 --- a/comfy/model_base.py +++ b/comfy/model_base.py @@ -300,6 +300,7 @@ class BaseModel(torch.nn.Module): free_cpu_memory = get_free_memory(torch.device("cpu")) logging.info(f"load model weights start, free cpu memory size {free_cpu_memory/(1024*1024*1024)} GB") + logging.info(f"model destination device {next(self.diffusion_model.parameters()).device}") to_load = self.model_config.process_unet_state_dict(to_load) logging.info(f"load model {self.model_config} weights process end") m, u = self.diffusion_model.load_state_dict(to_load, strict=False)