diff --git a/README.md b/README.md index b530bd9..760f6c0 100644 --- a/README.md +++ b/README.md @@ -2,8 +2,8 @@ ## Overview -Welcome! I've developed a set of custom nodes for ComfyUI that allows you to use Core ML models in your ComfyUI -workflows. +Welcome! In this repository you'll find a set of custom nodes for ComfyUI that allows you to use Core ML models in your +ComfyUI workflows. These models are designed to leverage the Apple Neural Engine (ANE) on Apple Silicon (M1/M2) machines, thereby enhancing your workflows and improving performance. @@ -121,10 +121,6 @@ node is a `coreml_model` object that can be used with the Core ML Sampler. - **Outputs**: - **coreml_model**: A Core ML model that can be used with the Core ML Sampler. -> [!NOTE] -> Some models are designed to support ControlNet. If you're using such a model, -> make sure to provide a ControlNet input; otherwise, the model will use random noise as ControlNet input. - #### Core ML Sampler (`CoreMLSampler`) ![CoreMLSampler](./assets/sampler.png?raw=true) diff --git a/coreml_suite/controlnet.py b/coreml_suite/controlnet.py index 48e0cb0..2e130dc 100644 --- a/coreml_suite/controlnet.py +++ b/coreml_suite/controlnet.py @@ -5,7 +5,6 @@ import numpy as np import torch from coreml_suite.latents import chunk_batch -from coreml_suite.logger import logger def expand_inputs(inputs):