Add more advanced LCM Sampler

__init__.py

@@ -4,12 +4,17 @@ import sys
 sys.path.append(os.path.dirname(__file__))
 
 from coreml_suite.nodes import CoreMLLoaderUNet, CoreMLSampler, CoreMLModelAdapter
-from coreml_suite.lcm import CoreMLConverterLCM, CoreMLSamplerLCM_Simple
+from coreml_suite.lcm import (
+    CoreMLSamplerLCM,
+    CoreMLConverterLCM,
+    CoreMLSamplerLCM_Simple,
+)
 
 NODE_CLASS_MAPPINGS = {
     "CoreMLUNetLoader": CoreMLLoaderUNet,
     "CoreMLSampler": CoreMLSampler,
     "CoreMLModelAdapter": CoreMLModelAdapter,
+    "Core ML LCM Sampler": CoreMLSamplerLCM,
     "Core ML LCM Sampler (Simple)": CoreMLSamplerLCM_Simple,
     "CoreMLConverterLCM": CoreMLConverterLCM,
 }
@@ -17,6 +22,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
     "CoreMLUNetLoader": "Load Core ML UNet",
     "CoreMLSampler": "Core ML Sampler",
     "CoreMLModelAdapter": "Core ML Adapter (Experimental)",
+    "Core ML LCM Sampler": "Core ML LCM Sampler",
     "Core ML LCM Sampler (Simple)": "Core ML LCM Sampler (Simple)",
     "CoreMLConverterLCM": "Convert LCM to Core ML",
 }