From 6172e16572369879778c4a5a0f051de17c16283d Mon Sep 17 00:00:00 2001 From: Quentin HENRY Date: Tue, 22 Apr 2025 17:29:41 +0200 Subject: [PATCH] faster model loading for weight update --- modules/sd_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 55bd9ca5e..a0f2e3e49 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -961,7 +961,7 @@ def reload_model_weights(sd_model=None, info=None, forced_reload=False): if sd_model is not None: sd_unet.apply_unet("None") - send_model_to_cpu(sd_model) + send_model_to_device(sd_model) sd_hijack.model_hijack.undo_hijack(sd_model) state_dict = get_checkpoint_state_dict(checkpoint_info, timer)