From 8f7f0542ca71264d384a98c736816254ff90465a Mon Sep 17 00:00:00 2001 From: Vaibhav Bhembre Date: Thu, 8 Aug 2024 15:56:57 -0400 Subject: [PATCH] benchmarking: fix the variable to point to the correct one --- examples/benchmark_buffer_size/benchmark.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/benchmark_buffer_size/benchmark.py b/examples/benchmark_buffer_size/benchmark.py index af6afb14..f18d9ae1 100755 --- a/examples/benchmark_buffer_size/benchmark.py +++ b/examples/benchmark_buffer_size/benchmark.py @@ -132,7 +132,7 @@ model_name: str = args.model -http_uri = f"http://{args.bucket}.{args.s3_url}/{model_name}/model.tensors" +http_uri = f"http://{args.bucket}.{args.s3_endpoint}/{model_name}/model.tensors" https_uri = http_uri.replace("http://", "https://") s3_uri = f"s3://{args.bucket}/{model_name}/model.tensors" @@ -140,7 +140,7 @@ file_uri = f"{args.file_prefix}{sanitized_model_file}.tensors" local_uri = f"http://localhost:3000/{sanitized_model_file}.tensors" -s3_endpoint = f"http://{args.s3_url}" +s3_endpoint = f"http://{args.s3_endpoint}" if args.test_http: s3_endpoint = s3_endpoint.replace("http://", "https://")