Fix a typo in amd-llama model conversion.

haozha111 · copybara-github · commit fc6c8884a6a4 · 2025-02-20T09:32:39.000-08:00
PiperOrigin-RevId: 729143030
diff --git a/ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py b/ai_edge_torch/generative/examples/amd_llama_135m/convert_to_tflite.py
@@ -41,7 +41,7 @@
 )
 _OUTPUT_NAME_PREFIX = flags.DEFINE_string(
     'output_name_prefix',
-    'deepseek',
+    'amd_llama',
     'The prefix of the output tflite model name.',
 )
 _PREFILL_SEQ_LENS = flags.DEFINE_multi_integer(

Original file line number	Diff line number	Diff line change
`@@ -41,7 +41,7 @@`
`41`	`41`	`)`
`42`	`42`	`_OUTPUT_NAME_PREFIX = flags.DEFINE_string(`
`43`	`43`	`'output_name_prefix',`
`44`		`- 'deepseek',`
	`44`	`+ 'amd_llama',`
`45`	`45`	`'The prefix of the output tflite model name.',`
`46`	`46`	`)`
`47`	`47`	`_PREFILL_SEQ_LENS = flags.DEFINE_multi_integer(`