DominikDoom/a1111-sd-webui-tagcomplete
1
0
Fork 0
mirror of https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git synced 2026-01-26 19:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support png embeddings

Browse Source 
Fixes #33
This commit is contained in:
Dominik Reh
2022-10-24 14:01:44 +02:00
parent 4af8d5285d
commit e8cf50cdaa
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
javascript/tagAutocomplete.js
View File

@@ -599,7 +599,7 @@ onUiUpdate(function () {
try {
embeddings = readFile("file/tags/temp/emb.txt").split("\n")
.filter(x => x.trim().length > 0) // Remove empty lines
.map(x => x.replace(".bin", "").replace(".pt", "")); // Remove file extensions
.map(x => x.replace(".bin", "").replace(".pt", "").replace(".png", "")); // Remove file extensions
} catch (e) {
console.error("Error loading embeddings.txt: " + e);
}

2
scripts/tag_autocomplete_helper.py
View File

@@ -20,7 +20,7 @@ def get_wildcards():
def get_embeddings():
"""Returns a list of all embeddings"""
return [str(e.relative_to(EMB_PATH)) for e in EMB_PATH.glob("**/*") if e.suffix in {".bin", ".pt"}]
return [str(e.relative_to(EMB_PATH)) for e in EMB_PATH.glob("**/*") if e.suffix in {".bin", ".pt", ".png"}]
def write_to_temp_file(name, data):