Merge pull request #2 from jordanramstad/prompt-fix

fix prompt
2026-04-23 03:00:31 -04:00 · 2023-02-20 20:11:00 -08:00
parent de89041779 3732af63e8
commit 8f527c2b2d
1 changed files with 4 additions and 3 deletions
--- 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,