add alt loaders for checkpoint and lora

Comfy-Org · Sep 26, 2024 · 2a8c8b7 · 2a8c8b7
1 parent e5ed977
commit 2a8c8b7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/stores/modelToNodeStore.ts b/src/stores/modelToNodeStore.ts
@@ -77,7 +77,13 @@ export const useModelToNodeStore = defineStore('modelToNode', {
       }
       this.haveDefaultsLoaded = true
       this.quickRegister('checkpoints', 'CheckpointLoaderSimple', 'ckpt_name')
+      this.quickRegister(
+        'checkpoints',
+        'ImageOnlyCheckpointLoader',
+        'ckpt_name'
+      )
       this.quickRegister('loras', 'LoraLoader', 'lora_name')
+      this.quickRegister('loras', 'LoraLoaderModelOnly', 'lora_name')
       this.quickRegister('vae', 'VAELoader', 'vae_name')
       this.quickRegister('controlnet', 'ControlNetLoader', 'control_net_name')
     }