From 3df0fb8008b58d5cfe4009d1bf9148b422187273 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Mon, 4 Nov 2024 12:42:16 -0500 Subject: [PATCH] apply balanced to recursive pipe Signed-off-by: Vladimir Mandic --- modules/sd_models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/modules/sd_models.py b/modules/sd_models.py index 575e9318a..1a17660a3 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -416,6 +416,8 @@ def apply_balanced_offload(sd_model): devices.torch_gc(fast=True) apply_balanced_offload_to_module(sd_model) + if hasattr(sd_model, "pipe"): + apply_balanced_offload_to_module(sd_model.pipe) if hasattr(sd_model, "prior_pipe"): apply_balanced_offload_to_module(sd_model.prior_pipe) if hasattr(sd_model, "decoder_pipe"):