From a652c55bf77030deeea3f16e8768b84298a2e2e3 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Mon, 1 Dec 2025 12:50:22 +0200
Subject: [PATCH] Remove LoRAs when fully offloading as well

---
 nodes_sampler.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nodes_sampler.py b/nodes_sampler.py
index 8112e8f..7e4ff44 100644
--- a/nodes_sampler.py
+++ b/nodes_sampler.py
@@ -68,6 +68,7 @@ def offload_transformer(transformer):
                 setattr(module, attr_name, MetaParameter(param.data.dtype, quant_type))
             else:
                 pass
+        remove_lora_from_module(transformer)
     else:
         transformer.to(offload_device)