From d3dff5df151f5ca6b12911df9318b1667d9cfa1a Mon Sep 17 00:00:00 2001 From: vladmandic Date: Wed, 4 Feb 2026 15:05:22 +0100 Subject: [PATCH] one-more-offload Signed-off-by: vladmandic --- modules/sd_models.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/modules/sd_models.py b/modules/sd_models.py index 9d93ebe96..a88acded9 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -1245,6 +1245,12 @@ def set_diffuser_pipe(pipe, new_pipe_type): fn = f'{sys._getframe(2).f_code.co_name}:{sys._getframe(1).f_code.co_name}' # pylint: disable=protected-access shared.log.debug(f"Pipeline class change: original={cls} target={new_pipe.__class__.__name__} device={pipe.device} fn={fn}") # pylint: disable=protected-access + + if shared.opts.diffusers_offload_mode == 'none': + move_model(new_pipe, pipe.device) + else: + set_diffuser_offload(new_pipe, op='model') + pipe = new_pipe return pipe