From 73b13e0d2346c3c14f04aa4afc1d3605d26ee793 Mon Sep 17 00:00:00 2001 From: Chris Chance Date: Wed, 29 Nov 2023 20:46:04 -0500 Subject: [PATCH] Update ModelSamplingDiscreteLCM to Distilled for latest comfyui --- coreml_suite/lcm/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/coreml_suite/lcm/utils.py b/coreml_suite/lcm/utils.py index fbd7c5b..c559fde 100644 --- a/coreml_suite/lcm/utils.py +++ b/coreml_suite/lcm/utils.py @@ -1,7 +1,7 @@ import torch from comfy.model_management import get_torch_device -from comfy_extras.nodes_model_advanced import ModelSamplingDiscreteLCM, LCM +from comfy_extras.nodes_model_advanced import ModelSamplingDiscreteDistilled, LCM def is_lcm(coreml_model): @@ -42,7 +42,7 @@ def model_function_wrapper(w_embedding): def lcm_patch(model): m = model.clone() sampling_type = LCM - sampling_base = ModelSamplingDiscreteLCM + sampling_base = ModelSamplingDiscreteDistilled class ModelSamplingAdvanced(sampling_base, sampling_type): pass