mirror of https://github.com/google/gemma.cpp.git
Fix a problem in run_example.py
PiperOrigin-RevId: 767017932
This commit is contained in:
parent
9efdcfd45c
commit
9f74a1a098
|
|
@ -48,7 +48,6 @@ def main(argv: Sequence[str]) -> None:
|
||||||
model = gemma.GemmaModel(
|
model = gemma.GemmaModel(
|
||||||
tokenizer_path=tokenizer_path,
|
tokenizer_path=tokenizer_path,
|
||||||
weights_path=weights_path,
|
weights_path=weights_path,
|
||||||
model_flag="gemma2-2b-it",
|
|
||||||
max_threads=24,
|
max_threads=24,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -87,7 +86,6 @@ def main(argv: Sequence[str]) -> None:
|
||||||
model = gemma.GemmaModel(
|
model = gemma.GemmaModel(
|
||||||
tokenizer_path=tokenizer_path,
|
tokenizer_path=tokenizer_path,
|
||||||
weights_path=weights_path,
|
weights_path=weights_path,
|
||||||
model_flag="paligemma-224",
|
|
||||||
max_threads=24,
|
max_threads=24,
|
||||||
)
|
)
|
||||||
image = np.array(
|
image = np.array(
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue