Skip to content

Commit

Permalink
Add support to export XNNPACK based static_llama
Browse files Browse the repository at this point in the history
Differential Revision: D67867190

Pull Request resolved: #7535
  • Loading branch information
derekxu authored Jan 7, 2025
1 parent 507c767 commit a29dc49
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion examples/models/llama/export_llama_lib.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,14 @@
verbosity_setting = None


EXECUTORCH_DEFINED_MODELS = ["stories110m", "llama2", "llama3", "llama3_1", "llama3_2"]
EXECUTORCH_DEFINED_MODELS = [
"stories110m",
"llama2",
"llama3",
"llama3_1",
"llama3_2",
"static_llama",
]
TORCHTUNE_DEFINED_MODELS = ["llama3_2_vision"]


Expand Down

0 comments on commit a29dc49

Please sign in to comment.