Add exportable baby llama example (#4345)
Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/4345 Add a small LLaMa model, based on the babyllama paper. Note that this test case is only one layer by default, and the number of layers can be adjusted in the test. Removed some pyre changes that broke the OSS AoT export, and added some required passes and operators. Reviewed By: dulinriley Differential Revision: D60073137 fbshipit-source-id: 8379296ad0aa4099b09d033b33479165d7c7c5c9
M
Matthias Cremon committed
1e143339d463262694244c56f2fdb698ec28e3c5
Parent: db1c4d8
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com>
on 7/30/2024, 2:44:46 AM