up

metascroy · metascroy · commit 96dc88ee61a9 · 2025-09-29T17:07:27.000-07:00
diff --git a/examples/models/llama/export_llama_lib.py b/examples/models/llama/export_llama_lib.py
@@ -757,7 +757,7 @@ def _prepare_for_llama_export(llm_config: LlmConfig) -> LLMEdgeManager:
             preq_embedding_quantize=llm_config.base.preq_embedding_quantize,
             local_global_attention=llm_config.model.local_global_attention,
             use_torchao_kernels_linear=llm_config.backend.torchao.use_torchao_kernels_linear,
-            use_torchao_kernels_tied_embedding=llm_config.backend.torchao.convert_tied_embedding,
+            use_torchao_kernels_tied_embedding=llm_config.backend.torchao.use_torchao_kernels_tied_embedding,
         )
     )
 

Original file line number	Diff line number	Diff line change
`@@ -757,7 +757,7 @@ def _prepare_for_llama_export(llm_config: LlmConfig) -> LLMEdgeManager:`
`757`	`757`	`preq_embedding_quantize=llm_config.base.preq_embedding_quantize,`
`758`	`758`	`local_global_attention=llm_config.model.local_global_attention,`
`759`	`759`	`use_torchao_kernels_linear=llm_config.backend.torchao.use_torchao_kernels_linear,`
`760`		`- use_torchao_kernels_tied_embedding=llm_config.backend.torchao.convert_tied_embedding,`
	`760`	`+ use_torchao_kernels_tied_embedding=llm_config.backend.torchao.use_torchao_kernels_tied_embedding,`
`761`	`761`	`)`
`762`	`762`	`)`
`763`	`763`