fix phi-3 conversion ckpt path.

haozha111 · copybara-github · commit 58f1c719bdf3 · 2025-01-09T15:17:49.000-08:00
PiperOrigin-RevId: 713811997
diff --git a/ai_edge_torch/generative/examples/phi/convert_phi3_to_tflite.py b/ai_edge_torch/generative/examples/phi/convert_phi3_to_tflite.py
@@ -26,7 +26,7 @@
 
 _CHECKPOINT_PATH = flags.DEFINE_string(
     'checkpoint_path',
-    os.path.join(pathlib.Path.home(), 'Downloads/llm_data/gemma-2b'),
+    os.path.join(pathlib.Path.home(), 'Downloads/llm_data/phi3'),
     'The path to the model checkpoint, or directory holding the checkpoint.',
 )
 _OUTPUT_PATH = flags.DEFINE_string(

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`
`27`	`27`	`_CHECKPOINT_PATH = flags.DEFINE_string(`
`28`	`28`	`'checkpoint_path',`
`29`		`- os.path.join(pathlib.Path.home(), 'Downloads/llm_data/gemma-2b'),`
	`29`	`+ os.path.join(pathlib.Path.home(), 'Downloads/llm_data/phi3'),`
`30`	`30`	`'The path to the model checkpoint, or directory holding the checkpoint.',`
`31`	`31`	`)`
`32`	`32`	`_OUTPUT_PATH = flags.DEFINE_string(`