From 41759f5e67d93d116ca7e22d3c3489c5cc4e0e6a Mon Sep 17 00:00:00 2001 From: Will Miao Date: Thu, 25 Dec 2025 21:15:28 +0800 Subject: [PATCH] refactor: use `model.clone()` instead of `copy.deepcopy()` for model duplication, see #733 --- py/nodes/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/nodes/utils.py b/py/nodes/utils.py index 5ea94265..402025e1 100644 --- a/py/nodes/utils.py +++ b/py/nodes/utils.py @@ -103,7 +103,7 @@ def nunchaku_load_lora(model, lora_name, lora_strength): # Save the transformer temporarily model_wrapper.model = None - ret_model = copy.deepcopy(model) # copy everything except the model + ret_model = model.clone() ret_model_wrapper = ret_model.model.diffusion_model # Restore the model and set it for the copy