From 7859fd22a0bea4a74072f5078a72c4c04195dad9 Mon Sep 17 00:00:00 2001
From: AI-Casanova <54461896+AI-Casanova@users.noreply.github.com>
Date: Sun, 6 Aug 2023 22:33:27 -0500
Subject: [PATCH] Unload LoRA on early stoppage

---
 modules/processing_diffusers.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/processing_diffusers.py b/modules/processing_diffusers.py
index 15364b739..ccbaed710 100644
--- a/modules/processing_diffusers.py
+++ b/modules/processing_diffusers.py
@@ -132,6 +132,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
     # parsed_prompt = [parse_prompt_attention(prompt) for prompt in prompts]
 
     if shared.state.interrupted or shared.state.skipped:
+        unload_diffusers_lora()
         return results
 
     if shared.opts.diffusers_move_base and not shared.sd_model.has_accelerate:
@@ -154,6 +155,7 @@ def process_diffusers(p: StableDiffusionProcessing, seeds, prompts, negative_pro
     output = shared.sd_model(**pipe_args) # pylint: disable=not-callable
 
     if shared.state.interrupted or shared.state.skipped:
+        unload_diffusers_lora()
         return results
 
     if shared.sd_refiner is None or not p.enable_hr: