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 e32ea56 commit 11b2f0fCopy full SHA for 11b2f0f
examples/models/llama/export_llama_lib.py
@@ -29,6 +29,7 @@
29
30
from executorch.devtools.etrecord import generate_etrecord as generate_etrecord_func
31
32
+from executorch.examples.models.llama.config.llm_config import LlmConfig
33
from executorch.examples.models.llama.config.llm_config_utils import (
34
convert_args_to_llm_config,
35
)
@@ -1198,6 +1199,7 @@ def _load_llama_model(
1198
1199
use_qnn: bool = False,
1200
export_only: bool = False,
1201
args,
1202
+ llm_config: Optional[LlmConfig] = None,
1203
) -> "LLMEdgeManager":
1204
"""
1205
A helper util that builds a Llama2 model. It returns a LLMEdgeManager that
0 commit comments