From f3e64b1fa5ae9757cc67b8eb198b01e76a6c3c7c Mon Sep 17 00:00:00 2001 From: Symbiomatrix Date: Sun, 3 Mar 2024 13:47:39 +0200 Subject: [PATCH] Update tag_autocomplete_helper.py --- scripts/tag_autocomplete_helper.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/scripts/tag_autocomplete_helper.py b/scripts/tag_autocomplete_helper.py index b3a7b47..d8605d0 100644 --- a/scripts/tag_autocomplete_helper.py +++ b/scripts/tag_autocomplete_helper.py @@ -186,15 +186,15 @@ def get_embeddings(sd_model): continue if emb.shape is None: - emb_unknown.append((Path(emb.filename), key, "")) + emb_unknown.append((Path(emb.filename), Path(emb.filename).relative_to(EMB_PATH).as_posix(), "")) elif emb.shape == V1_SHAPE: - emb_v1.append((Path(emb.filename), key, "v1")) + emb_v1.append((Path(emb.filename), Path(emb.filename).relative_to(EMB_PATH).as_posix(), "v1")) elif emb.shape == V2_SHAPE: - emb_v2.append((Path(emb.filename), key, "v2")) + emb_v2.append((Path(emb.filename), Path(emb.filename).relative_to(EMB_PATH).as_posix(), "v2")) elif emb.shape == VXL_SHAPE: - emb_vXL.append((Path(emb.filename), key, "vXL")) + emb_vXL.append((Path(emb.filename), Path(emb.filename).relative_to(EMB_PATH).as_posix(), "vXL")) else: - emb_unknown.append((Path(emb.filename), key, "")) + emb_unknown.append((Path(emb.filename), Path(emb.filename).relative_to(EMB_PATH).as_posix(), "")) results = sort_models(emb_v1) + sort_models(emb_v2) + sort_models(emb_vXL) + sort_models(emb_unknown) except AttributeError: