Coreml prefill

[metascroy]

Export to CoreML:

export MODEL_IN=$HOME/stories110M/stories110M.pt                                                                                          
export TOKENIZER=$HOME/stories110M/tokenizer.bin
export PARAMS=$HOME/stories110M/params.json
export MODEL_OUT_COREML=$HOME/stories110M/llama_model_coreml.pte

python -m examples.models.llama.export_llama --disable_dynamic_shape --coreml -c $MODEL_IN -p $PARAMS --coreml-ios 18 --output_name=$MODEL_OUT_COREML -E "4,32" --max_seq_length 512 --verbose &> log.txt

Extract CoreML mlpackage:

python examples/apple/coreml/scripts/extract_coreml_models.py -m $MODEL_OUT_COREML

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/6895

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

• [DomainsOnly] Jobs fail with GLIBC version not found

❌ 3 New Failures

As of commit b4db52f with merge base 92ee522 ():

NEW FAILURES - The following jobs have failed:

• Check Labels / Check labels (gh)
  RuntimeError: Error checking labels: PR does not have required labels
• Lint / lintrunner / linux-job (gh)
  >>> Lint for extension/llm/export/partitioner_lib.py:
• pull / test-llama-runner-qnn-linux (fp32, qnn) / linux-job (gh)
  RuntimeError: Command docker exec -t aa274721a23f622afe23027ac0bf76fa3ac7eb803117afba512a19bf716d3d57 /exec failed with exit code 92

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[facebook-github-bot]

@metascroy has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[github-actions]

This PR needs a release notes: label

If your changes are user facing and intended to be a part of release notes, please use a label starting with release notes:.

If not, please add the topic: not user facing label.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "topic: not user facing"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.

[facebook-github-bot]

@metascroy has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.

[github-actions]

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.