Skip to content

Commit 68c39b3

Browse files
committed
Default chunking arg for --merge-chunks-in-pipeline-model when called from torch2coreml
1 parent c69deb7 commit 68c39b3

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python_coreml_stable_diffusion/torch2coreml.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -905,6 +905,7 @@ def convert_unet(pipe, args, model_name = None):
905905
logger.info(f"Chunking {model_name} in two approximately equal MLModels")
906906
args.mlpackage_path = out_path
907907
args.remove_original = False
908+
args.merge_chunks_in_pipeline_model = False
908909
chunk_mlprogram.main(args)
909910

910911

0 commit comments

Comments
 (0)