From 3732af63e8cf73c28cab7478d783cd7e347fefbd Mon Sep 17 00:00:00 2001
From: neecapp <ryree0@gmail.com>
Date: Mon, 20 Feb 2023 23:06:05 -0500
Subject: [PATCH] fix prompt

---
 ldm/generate.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/ldm/generate.py b/ldm/generate.py
index c1981d2e23..a8d13f3a1f 100644
--- a/ldm/generate.py
+++ b/ldm/generate.py
@@ -513,6 +513,10 @@ class Generate:
         except AttributeError:
             pass
 
+        # lora MUST process prompt before conditioning
+        if self.model.lora_manager:
+            self.model.lora_manager.load_lora()
+
         try:
             uc, c, extra_conditioning_info = get_uc_and_c_and_ec(
                 prompt,
@@ -549,9 +553,6 @@ class Generate:
                 else None
             )
 
-            if self.model.lora_manager:
-                self.model.lora_manager.load_lora()
-
             results = generator.generate(
                 prompt,
                 iterations=iterations,