benchmarking: fix the variable to point to the correct one

coreweave · Aug 8, 2024 · 8f7f054 · 8f7f054
1 parent 44a3e60
commit 8f7f054
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/benchmark_buffer_size/benchmark.py b/examples/benchmark_buffer_size/benchmark.py
@@ -132,15 +132,15 @@
 
 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"
 sanitized_model_file = model_name.replace("/", "_")
 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://")