From e6ed5feb884426badece8d98f37b95edc5140871 Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Fri, 23 Aug 2024 14:18:38 +0800 Subject: [PATCH] Update gradio_utils.py --- utils/gradio_utils.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/utils/gradio_utils.py b/utils/gradio_utils.py index 40f9457..36b9c2f 100644 --- a/utils/gradio_utils.py +++ b/utils/gradio_utils.py @@ -439,10 +439,8 @@ def is_torch2_available(): def character_to_dict(general_prompt,lora,add_trigger_words): character_dict = {} generate_prompt_arr = general_prompt.splitlines() - if lora!="none": - if lora in lora_lightning_list: - pass - else: + if lora: + if lora not in lora_lightning_list: generate_prompt_arr = [item + add_trigger_words for item in generate_prompt_arr] #print(prompts) character_index_dict = {}