Skip to content

Commit c6884bf

Browse files
committed
add some logging
1 parent 3fc577c commit c6884bf

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python_coreml_stable_diffusion/chunk_mlprogram.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -355,12 +355,14 @@ def main(args):
355355
# Starting from coremltools==8.0b2, there is this `bisect_model` API that
356356
# we can directly call into.
357357
from coremltools.models.utils import bisect_model
358+
logger.info(f"Start chunking model {args.mlpackage_path} into two pieces.")
358359
ct.models.utils.bisect_model(
359360
model=args.mlpackage_path,
360361
output_dir=args.o,
361362
merge_chunks_to_pipeline=args.merge_chunks_in_pipeline_model,
362363
check_output_correctness=args.check_output_correctness,
363364
)
365+
logger.info(f"Model chunking is done.")
364366

365367
# Remove original (non-chunked) model if requested
366368
if args.remove_original:

0 commit comments

Comments
 (0)