1580158
/
model-keyword
mirror of https://github.com/mix1009/model-keyword.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

sort lora models.

pull/33/head
ChunKoo Park 2023-03-03 13:48:13 +09:00
parent 8d1e2cd0a1
commit 50d53d836d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/model_keyword.py
View File

@ -247,7 +247,7 @@ class Script(scripts.Script):
def get_embeddings():
return [os.path.basename(x) for x in glob.glob(f'{shared.cmd_opts.embeddings_dir}/*.pt')]
def get_loras():
return list(find_files(shared.cmd_opts.lora_dir,['safetensors','ckpt','pt']))
return sorted(list(find_files(shared.cmd_opts.lora_dir,['safetensors','ckpt','pt'])), key=str.casefold)
# return [os.path.basename(x) for x in glob.glob(f'{shared.cmd_opts.lora_dir}/*.safetensors')]
def update_keywords():