Update ModelSamplingDiscreteLCM to Distilled for latest comfyui

2024-06-28 15:52:54 +02:00
parent 991ab6a40a
commit 73b13e0d23
1 changed files with 2 additions and 2 deletions
@@ -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