Skip to content

Commit d0d05ce

Browse files
committed
Fix compute unit for chunk_mlprogram parity checks
1 parent f3a2124 commit d0d05ce

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python_coreml_stable_diffusion/chunk_mlprogram.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -321,7 +321,7 @@ def main(args):
321321

322322
# Save and reload to ensure CPU placement
323323
pipeline_model.save(out_path_pipeline)
324-
pipeline_model = ct.models.MLModel(out_path_pipeline)
324+
pipeline_model = ct.models.MLModel(out_path_pipeline, compute_units=ct.ComputeUnit.CPU_ONLY)
325325

326326
if args.check_output_correctness:
327327
logger.info("Verifying output correctness of pipeline model")

0 commit comments

Comments
 (0)