From e2b6dfeeb9af7dbc43821e3b5cafacf3db1da7fb Mon Sep 17 00:00:00 2001 From: Jordan Date: Mon, 20 Feb 2023 21:33:20 -0700 Subject: [PATCH] Update generate.py --- ldm/generate.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/ldm/generate.py b/ldm/generate.py index a8d13f3a1f..8e3b27b2ac 100644 --- a/ldm/generate.py +++ b/ldm/generate.py @@ -482,6 +482,8 @@ class Generate: if self.model.lora_manager: prompt = self.model.lora_manager.configure_prompt(prompt) + # lora MUST process prompt before conditioning + self.model.lora_manager.load_lora() # apply the concepts library to the prompt prompt = self.huggingface_concepts_library.replace_concepts_with_triggers( @@ -513,10 +515,6 @@ 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,