From 54391300b16a2925ad4bed7e793db1bf5d3a7e2d Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 3 Sep 2025 10:00:51 +0300 Subject: [PATCH] Never use main_device with unmerged loras --- nodes_model_loading.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/nodes_model_loading.py b/nodes_model_loading.py index d955e9f..221b5e6 100644 --- a/nodes_model_loading.py +++ b/nodes_model_loading.py @@ -990,7 +990,9 @@ class WanVideoModelLoader: raise ValueError("GGUF models do not support LoRA merging, please disable merge_loras in the LoRA select node.") transformer_load_device = device if load_device == "main_device" else offload_device - + if lora is not None and not merge_loras: + transformer_load_device = offload_device + base_dtype = {"fp8_e4m3fn": torch.float8_e4m3fn, "fp8_e4m3fn_fast": torch.float8_e4m3fn, "bf16": torch.bfloat16, "fp16": torch.float16, "fp16_fast": torch.float16, "fp32": torch.float32}[base_precision] if base_precision == "fp16_fast":