TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed note parse bug.

Browse Source 
- Avoid adding empty string to list of activation text.
This commit is contained in:
Christian Bastian
2024-09-21 18:33:52 -04:00
parent 4757978164
commit 7a49b548a2
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
__init__.py
View File

@@ -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: 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} # {'description': str, 'sd version': str, 'activation text': str, 'preferred weight': int, 'notes': str}
model_description = model_version_info.get("description") model_description = model_version_info.get("description")
model_trigger_words = model_version_info.get("activation text") activation_text = model_version_info.get("activation text")
if type(model_trigger_words) is str: if activation_text != "":
model_trigger_words = [model_trigger_words] model_trigger_words = [activation_text]
else: else:
model_trigger_words = [] model_trigger_words = []
model_version_description = model_version_info.get("notes") model_version_description = model_version_info.get("notes")
else: else:
return "" 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_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_version_description = model_version_description if model_version_description is not None else ""
model_name = model_name if type(model_name) is str else "Model Description" model_name = model_name if model_name is not None else "Model Description"
notes = "" notes = ""
if len(model_trigger_words) > 0: if len(model_trigger_words) > 0: