diff --git a/README.md b/README.md index a7eb693..b46de1b 100644 --- a/README.md +++ b/README.md @@ -193,8 +193,8 @@ the node name, so if the model already exists, the node will not convert it agai - **ckpt_name**: The name of the checkpoint to convert. This should be the name of the checkpoint file stored in the `models/checkpoints` directory. - **model_version**: Whether the model is based on SD1.5 or SDXL. - - **height**: The desired height of the image generated by the model. The default is 512. Must be a multiple of 8. - - **width**: The desired width of the image generated by the model. The default is 512. Must be a multiple of 8. + - **height**: The desired height of the image generated by the model. The default is 512. Any positive multiple of 8 is accepted. + - **width**: The desired width of the image generated by the model. The default is 512. Any positive multiple of 8 is accepted. - **batch_size**: The batch size of generated images. If you're planning to generate batches of images, you can try increasing this value to speed up the generation process. The default is 1. - **attention_implementation**: The attention implementation used when converting the model. Choose SPLIT_EINSUM or diff --git a/coreml_suite/nodes.py b/coreml_suite/nodes.py index 9134950..f6d9301 100644 --- a/coreml_suite/nodes.py +++ b/coreml_suite/nodes.py @@ -225,8 +225,8 @@ class CoreMLConverter(COREML_NODE): ModelVersion.SDXL.name, ], ), - "height": ("INT", {"default": 512, "min": 256, "max": 2048, "step": 8}), - "width": ("INT", {"default": 512, "min": 256, "max": 2048, "step": 8}), + "height": ("INT", {"default": 512, "min": 8, "step": 8}), + "width": ("INT", {"default": 512, "min": 8, "step": 8}), "batch_size": ("INT", {"default": 1, "min": 1, "max": 64}), "attention_implementation": ( list(ATTENTION_IMPLEMENTATIONS), diff --git a/pyproject.toml b/pyproject.toml index 4d72d72..94b39e6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-coremlsuite" description = "This extension contains a set of custom nodes for ComfyUI that allow you to use Core ML models in your ComfyUI workflows." -version = "2.0.1" +version = "2.0.2" license = "MIT" requires-python = ">=3.12,<3.13" packages = [{ include = "coreml_suite" }] diff --git a/uv.lock b/uv.lock index 673a37c..35d7415 100644 --- a/uv.lock +++ b/uv.lock @@ -206,7 +206,7 @@ wheels = [ [[package]] name = "comfyui-coremlsuite" -version = "2.0.1" +version = "2.0.2" source = { virtual = "." } dependencies = [ { name = "coremltools" },