From d612d1ffef506a1f34db058dc9aa6c76ca16545c Mon Sep 17 00:00:00 2001 From: aszc-dev Date: Fri, 3 Nov 2023 01:27:44 +0100 Subject: [PATCH] Adjust default values for LCM nodes --- coreml_suite/lcm/lcm_sampler.py | 1 + coreml_suite/lcm/nodes.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) 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,