Add node to load LoRAs

This commit is contained in:
aszc-dev
2023-11-17 22:55:13 +01:00
committed by aszc
parent da3a8e13d3
commit 42ff10cd43
2 changed files with 41 additions and 15 deletions
+3
View File
@@ -8,6 +8,7 @@ from coreml_suite.nodes import (
CoreMLSampler,
CoreMLModelAdapter,
COREML_CONVERT,
COREML_LOAD_LORA,
)
from coreml_suite.lcm import (
COREML_CONVERT_LCM,
@@ -17,6 +18,7 @@ NODE_CLASS_MAPPINGS = {
"CoreMLUNetLoader": CoreMLLoaderUNet,
"CoreMLSampler": CoreMLSampler,
"CoreMLModelAdapter": CoreMLModelAdapter,
"Core ML LoRA Loader": COREML_LOAD_LORA,
"Core ML Converter": COREML_CONVERT,
"Core ML LCM Converter": COREML_CONVERT_LCM,
}
@@ -24,6 +26,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"CoreMLUNetLoader": "Load Core ML UNet",
"CoreMLSampler": "Core ML Sampler",
"CoreMLModelAdapter": "Core ML Adapter (Experimental)",
"Core ML LoRA Loader": "Load LoRA to use with Core ML",
"Core ML Converter": "Convert Checkpoint to Core ML",
"Core ML LCM Converter": "Convert LCM to Core ML",
}