feat: added lora name to output

This commit is contained in:
Dijkstra
2024-04-09 17:05:22 +02:00
parent 9d8b79844f
commit 96a69f7cbe
2 changed files with 19 additions and 17 deletions

View File

@@ -217,8 +217,10 @@ def parse_selector(selector, tags_list):
def append_lora_name_if_empty(tags_list, lora_path, enabled):
if not enabled or len(tags_list) > 0:
return tags_list
tags_list.append(get_lora_name_from_lora_path(lora_path))
return tags_list
def get_lora_name_from_lora_path(lora_path):
filename = os.path.splitext(lora_path)[0]
filename = os.path.basename(filename)
tags_list.append(filename)
return tags_list
return filename