rixty
/
SillyTavern
mirror of https://github.com/SillyTavern/SillyTavern.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add diffusion unets to ComfyUI models list 

rm string interpolation
This commit is contained in:
ceruleandeep 2024-11-18 22:17:19 +11:00
parent b46dae0d6e
commit bcf127e387
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/endpoints/stable-diffusion.js
View File

@ -411,10 +411,11 @@ comfy.post('/models', jsonParser, async (request, response) => {
const data = await result.json(); const data = await result.json();
const ckpts = data.CheckpointLoaderSimple.input.required.ckpt_name[0].map(it => ({ value: it, text: it })) || []; const ckpts = data.CheckpointLoaderSimple.input.required.ckpt_name[0].map(it => ({ value: it, text: it })) || [];
const unets = data.UNETLoader.input.required.unet_name[0].map(it => ({ value: it, text: `UNet: ${it}` })) || [];
// load list of GGUF unets from diffusion_models if the loader node is available // load list of GGUF unets from diffusion_models if the loader node is available
const ggufs = data.UnetLoaderGGUF?.input.required.unet_name[0].map(it => ({ value: it, text: `GGUF: ${it}` })) || []; const ggufs = data.UnetLoaderGGUF?.input.required.unet_name[0].map(it => ({ value: it, text: `GGUF: ${it}` })) || [];
const models = ckpts.concat(ggufs); const models = [...ckpts, ...unets, ...ggufs];
// make the display names of the models somewhat presentable // make the display names of the models somewhat presentable
models.forEach(it => it.text = it.text.replace(/\.[^.]*$/, '').replace(/_/g, ' ')); models.forEach(it => it.text = it.text.replace(/\.[^.]*$/, '').replace(/_/g, ' '));
@ -577,7 +578,7 @@ comfy.post('/generate', jsonParser, async (request, response) => {
return response.send(imgBuffer.toString('base64')); return response.send(imgBuffer.toString('base64'));
} catch (error) { } catch (error) {
console.log('ComfyUI error:', error); console.log('ComfyUI error:', error);
response.status(500).send(`${error.message}`); response.status(500).send(error.message);
return response; return response;
} }
}); });