From 7a49b548a20c891d4ec5b12e703cc164364cbc78 Mon Sep 17 00:00:00 2001 From: Christian Bastian <80225746+cdb-boop@users.noreply.github.com> Date: Sat, 21 Sep 2024 18:33:52 -0400 Subject: [PATCH] Fixed note parse bug. - Avoid adding empty string to list of activation text. --- __init__.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/__init__.py b/__init__.py index 6e0c59a..64198cf 100644 --- a/__init__.py +++ b/__init__.py @@ -324,18 +324,18 @@ class Civitai: elif "description" in model_version_info and "activation text" in model_version_info and "notes" in model_version_info: # {'description': str, 'sd version': str, 'activation text': str, 'preferred weight': int, 'notes': str} model_description = model_version_info.get("description") - model_trigger_words = model_version_info.get("activation text") - if type(model_trigger_words) is str: - model_trigger_words = [model_trigger_words] + activation_text = model_version_info.get("activation text") + if activation_text != "": + model_trigger_words = [activation_text] else: model_trigger_words = [] model_version_description = model_version_info.get("notes") else: return "" - model_description = model_description if type(model_description) is str else "" + model_description = model_description if model_description is not None else "" model_trigger_words = model_trigger_words if model_trigger_words is not None else [] - model_version_description = model_version_description if type(model_version_description) is str else "" - model_name = model_name if type(model_name) is str else "Model Description" + model_version_description = model_version_description if model_version_description is not None else "" + model_name = model_name if model_name is not None else "Model Description" notes = "" if len(model_trigger_words) > 0: