From 297d2fbbb98f34946465680562a547601b146224 Mon Sep 17 00:00:00 2001 From: senpai-a Date: Thu, 25 Jun 2026 23:56:24 +0900 Subject: [PATCH] feat/choose-diffuser-model --- src/ui.js | 20 ++++++++++++++++---- 1 file changed, 16 insertions(+), 4 deletions(-) diff --git a/src/ui.js b/src/ui.js index 81f6df3..1ac89cb 100644 --- a/src/ui.js +++ b/src/ui.js @@ -26,14 +26,26 @@ function saveSettings() { */ async function fetchCheckpoints() { const settings = getSettings(); + const ret = [] try { - const response = await fetch(`${settings.comfy_host}/object_info/CheckpointLoaderSimple`); - if (!response.ok) return []; - const data = await response.json(); - return data?.CheckpointLoaderSimple?.input?.required?.ckpt_name?.[0] ?? []; + const ckpt_response = await fetch(`${settings.comfy_host}/object_info/CheckpointLoaderSimple`); + if (ckpt_response.ok) { + const ckpt_data = await ckpt_response.json(); + ret.push( + ...(ckpt_data?.CheckpointLoaderSimple?.input?.required?.ckpt_name?.[0] ?? []) + ); + } + const unet_response = await fetch(`${settings.comfy_host}/object_info/UNETLoader`); + if (unet_response.ok) { + const unet_data = await unet_response.json(); + ret.push( + ...(unet_data?.UNETLoader?.input?.required?.unet_name?.[0] ?? []) + ); + } } catch (err) { return []; } + return ret } /**