github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

sort loras and TIs in case-insensitive fashion

Browse Source
This commit is contained in:
Lincoln Stein
2023-04-04 11:24:10 -04:00
parent ad5142d6f7
commit e3772f674d
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/invoke_ai_web_server.py
View File

@@ -535,7 +535,7 @@ class InvokeAIWebServer:
loras = loras + models
found_loras = []
for lora in sorted(loras):
for lora in sorted(loras, key=lambda s: s.stem.lower()):
location = str(lora.resolve()).replace("\\", "/")
found_loras.append({"name": lora.stem, "location": location})
socketio.emit("foundLoras", found_loras)
@@ -546,9 +546,9 @@ class InvokeAIWebServer:
def get_ti_triggers():
try:
local_triggers = self.generate.model.textual_inversion_manager.get_all_trigger_strings()
local_triggers = [{'name': x} for x in sorted(local_triggers)]
local_triggers = [{'name': x} for x in sorted(local_triggers, key=str.casefold)]
concepts = HuggingFaceConceptsLibrary().list_concepts(minimum_likes=6)
concepts = [{'name': f'<{x}>'} for x in sorted(concepts) if f'<{x}>' not in local_triggers]
concepts = [{'name': f'<{x}>'} for x in sorted(concepts, key=str.casefold) if f'<{x}>' not in local_triggers]
socketio.emit("foundTextualInversionTriggers", {'local_triggers': local_triggers, 'huggingface_concepts': concepts})
except Exception as e:
self.handle_exceptions(e)