Skip to content

Commit

Permalink
Add LLaMA model test (#4070)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #4070

As titled.

Required changes to make it work:
- replace `aten::full_like` with `aten::full`
- remove `aten::logical_neg` when used as an input to `aten::where`
- replace `-inf` value in `aten::full`

Reviewed By: JacobSzwejbka, cccclai

Differential Revision: D58974035

fbshipit-source-id: c9f8f9f83c4c9a92f84995290b08a74b672d959e
  • Loading branch information
mcremon-meta authored and facebook-github-bot committed Jun 27, 2024
1 parent 50baff1 commit 4157af4
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions examples/models/llama2/TARGETS
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ runtime.python_library(
base_module = "executorch.examples.models.llama2",
visibility = [
"//executorch/...",
"@EXECUTORCH_CLIENTS",
],
deps = [
"//caffe2:torch",
Expand Down

0 comments on commit 4157af4

Please sign in to comment.