Fix a problem in run_example.py

PiperOrigin-RevId: 767017932
This commit is contained in:
Daniel Keysers 2025-06-04 00:42:15 -07:00 committed by Copybara-Service
parent 9efdcfd45c
commit 9f74a1a098
1 changed files with 0 additions and 2 deletions

View File

@ -48,7 +48,6 @@ def main(argv: Sequence[str]) -> None:
model = gemma.GemmaModel(
tokenizer_path=tokenizer_path,
weights_path=weights_path,
model_flag="gemma2-2b-it",
max_threads=24,
)
@ -87,7 +86,6 @@ def main(argv: Sequence[str]) -> None:
model = gemma.GemmaModel(
tokenizer_path=tokenizer_path,
weights_path=weights_path,
model_flag="paligemma-224",
max_threads=24,
)
image = np.array(