diff --git a/coreml_suite/lcm/lcm_sampler.py b/coreml_suite/lcm/lcm_sampler.py index 984e725..3f0796e 100644 --- a/coreml_suite/lcm/lcm_sampler.py +++ b/coreml_suite/lcm/lcm_sampler.py @@ -18,6 +18,7 @@ class CoreMLSamplerLCM(CoreMLSampler): @classmethod def INPUT_TYPES(s): old_required = CoreMLSampler.INPUT_TYPES()["required"].copy() + old_required["steps"][1]["default"] = 4 old_required.pop("negative") old_required.pop("sampler_name") old_required.pop("scheduler") diff --git a/coreml_suite/lcm/nodes.py b/coreml_suite/lcm/nodes.py index 6172911..7e84339 100644 --- a/coreml_suite/lcm/nodes.py +++ b/coreml_suite/lcm/nodes.py @@ -15,7 +15,7 @@ class CoreMLConverterLCM: "required": { "height": ("INT", {"default": 512, "min": 512, "max": 768, "step": 8}), "width": ("INT", {"default": 512, "min": 512, "max": 768, "step": 8}), - "batch_size": ("INT", {"default": 4, "min": 1, "max": 64}), + "batch_size": ("INT", {"default": 1, "min": 1, "max": 64}), "compute_unit": ( [ ComputeUnit.CPU_AND_NE.name,