fix: embedding url

2023-10-06 17:00:55 +08:00
parent f4d75c56ab
commit ee89241011
1 changed files with 4 additions and 1 deletions
@@ -256,6 +256,9 @@ function showMyImageEditor(node) {
    (x) => x.name === "embedding_id"
  ).value;

+  const ckpt = node.widgets.find((x) => x.name === "ckpt").value;
+  const modelType = ckpt.match(/vit_[lbh]/)?.[0];
+
  const v = JSON.parse(
    node.widgets.find((x) => x.name === "image_prompts_json").value
  );
@@ -279,7 +282,7 @@ function showMyImageEditor(node) {
  );
  const embeedingUrl = api.apiURL(
    `/view?filename=${encodeURIComponent(
-      `${embeddingFilename}.npy`
+      `${embeddingFilename}_${modelType}.npy`
    )}&type=output&subfolder=`
  );
  loadNpyTensor(embeedingUrl).then((tensor) => {