fix typo in path name

premAI-io · Dec 8, 2023 · d1d1827 · d1d1827
1 parent 9c19b86
commit d1d1827
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bench_pytorch/bench.py b/bench_pytorch/bench.py
@@ -123,7 +123,7 @@ def benchmark(self, prompt: str, max_tokens: int, repetitions: int) -> None:
             f"Running Transformer benchmark (pytorch backend) on Llama with precision: {precision}"
         )
         llama_transformers_pytorch_benchmark = LlamaPyTorchBenchmark(
-            model_path=f"{args.models_dir}/llama2-7b-hf",
+            model_path=f"{args.models_dir}/llama-2-7b-hf",
             device=args.device,
             precision=precision,
         ).load_model()