We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3fc577c commit c6884bfCopy full SHA for c6884bf
python_coreml_stable_diffusion/chunk_mlprogram.py
@@ -355,12 +355,14 @@ def main(args):
355
# Starting from coremltools==8.0b2, there is this `bisect_model` API that
356
# we can directly call into.
357
from coremltools.models.utils import bisect_model
358
+ logger.info(f"Start chunking model {args.mlpackage_path} into two pieces.")
359
ct.models.utils.bisect_model(
360
model=args.mlpackage_path,
361
output_dir=args.o,
362
merge_chunks_to_pipeline=args.merge_chunks_in_pipeline_model,
363
check_output_correctness=args.check_output_correctness,
364
)
365
+ logger.info(f"Model chunking is done.")
366
367
# Remove original (non-chunked) model if requested
368
if args.remove_original:
0 commit comments