coremltools 6.0b1
Pre-release
Pre-release
- MLProgram compression: affine quantization, palettize, sparsify. See
coremltools.compression_utils
. - New options to set input and output types to multi array of type float16, grayscale image of type float16 and set output type as images, similar to the
coremltools.ImageType
used with inputs. - Support for PyTorch 1.11.0.
- Support for TensorFlow 2.8.
- [API Breaking Change] Remove
useCPUOnly
parameter fromcoremltools.convert
andcoremltools.models.MLModel
. Usecoremltools.ComputeUnit
instead. - Support for many new PyTorch and TensorFlow layers
- Many bug fixes and enhancements.
Known issues
- While conversion and CoreML models with Grayscale Float16 images should work with ios16/macos13 beta, the coremltools-CoreML python binding has an issue which would cause the
predict
API in coremltools to crash when the either the input or output is of type grayscale float16 - The new Compute unit configuration
MLComputeUnitsCPUAndNeuralEngine
is not available in coremltools yet