diff --git a/litert_torch/generative/examples/gemma3/README.md b/litert_torch/generative/examples/gemma3/README.md index 2a1e7fd02..3504dd4ec 100644 --- a/litert_torch/generative/examples/gemma3/README.md +++ b/litert_torch/generative/examples/gemma3/README.md @@ -57,8 +57,8 @@ config = bundler.BundleConfig( start_token=START_TOKEN, stop_tokens=STOP_TOKENS, output_filename="/tmp/gemma3.task", - prompt_prefix="user\n", - prompt_suffix="\nmodel\n", + prompt_prefix_user="user\n", + prompt_suffix_user="\nmodel\n", ) bundler.create_bundle(config) ``` diff --git a/litert_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py b/litert_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py index 4c37224b1..8fd39966c 100644 --- a/litert_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py +++ b/litert_torch/generative/examples/gemma3/convert_gemma3_to_tflite.py @@ -38,7 +38,9 @@ def main(_): else: raise ValueError(f'Unsupported model size: {_MODEL_SIZE.value}') - converter.build_and_convert_to_tflite_from_flags(model_builder) + output_name_prefix = f'gemma3-{_MODEL_SIZE.value}' + + converter.build_and_convert_to_tflite_from_flags(model_builder, output_name_prefix=output_name_prefix) if __name__ == '__main__':