Skip to content

Commit 11b2f0f

Browse files
committed
fix: add missing LlmConfig import and parameter in export_llama_lib.py
ghstack-source-id: e335b89 Pull Request resolved: #11167
1 parent e32ea56 commit 11b2f0f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

examples/models/llama/export_llama_lib.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929

3030
from executorch.devtools.etrecord import generate_etrecord as generate_etrecord_func
3131

32+
from executorch.examples.models.llama.config.llm_config import LlmConfig
3233
from executorch.examples.models.llama.config.llm_config_utils import (
3334
convert_args_to_llm_config,
3435
)
@@ -1198,6 +1199,7 @@ def _load_llama_model(
11981199
use_qnn: bool = False,
11991200
export_only: bool = False,
12001201
args,
1202+
llm_config: Optional[LlmConfig] = None,
12011203
) -> "LLMEdgeManager":
12021204
"""
12031205
A helper util that builds a Llama2 model. It returns a LLMEdgeManager that

0 commit comments

Comments
 (0)