diff --git a/src/stores/modelToNodeStore.ts b/src/stores/modelToNodeStore.ts index 3699678642..b22c6ab9b5 100644 --- a/src/stores/modelToNodeStore.ts +++ b/src/stores/modelToNodeStore.ts @@ -384,6 +384,15 @@ export const useModelToNodeStore = defineStore('modelToNode', () => { quickRegister('BiRefNet/pth', 'LS_LoadBiRefNetModel', 'model') quickRegister('onnx/human-parts', 'LS_HumanPartsUltra', '') quickRegister('lama', 'LaMa', 'lama_model') + + // CogVideoX video generation models (comfyui-cogvideoxwrapper) + quickRegister('CogVideo', 'DownloadAndLoadCogVideoModel', 'model') + + // Inpaint models (comfyui-inpaint-nodes) + quickRegister('inpaint', 'INPAINT_LoadInpaintModel', 'model_name') + + // LayerDiffuse transparent image generation (comfyui-layerdiffuse) + quickRegister('layer_model', 'LayeredDiffusionApply', 'config') } return {