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 f3a2124 commit d0d05ceCopy full SHA for d0d05ce
python_coreml_stable_diffusion/chunk_mlprogram.py
@@ -321,7 +321,7 @@ def main(args):
321
322
# Save and reload to ensure CPU placement
323
pipeline_model.save(out_path_pipeline)
324
- pipeline_model = ct.models.MLModel(out_path_pipeline)
+ pipeline_model = ct.models.MLModel(out_path_pipeline, compute_units=ct.ComputeUnit.CPU_ONLY)
325
326
if args.check_output_correctness:
327
logger.info("Verifying output correctness of pipeline model")
0 commit comments