Update bencmark model URLs to use S3 URL instead of alias

pytorch · Jun 5, 2024 · b11c473 · b11c473
1 parent b9a79fe
commit b11c473
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/benchmarks/models_config/bert_neuronx.yaml b/benchmarks/models_config/bert_neuronx.yaml
@@ -2,7 +2,7 @@
 bert_neuronx_batch_1:
     scripted_mode:
         benchmark_engine: "ab"
-        url: https://torchserve.pytorch.org/mar_files/BERTSeqClassification_torchscript_neuronx_batch_1.mar
+        url: https://torchserve.s3.amazonaws.com/mar_files/BERTSeqClassification_torchscript_neuronx_batch_1.mar
         workers:
             - 2
         batch_delay: 100
@@ -19,7 +19,7 @@ bert_neuronx_batch_1:
 bert_neuronx_batch_2:
     scripted_mode:
         benchmark_engine: "ab"
-        url: https://torchserve.pytorch.org/mar_files/BERTSeqClassification_torchscript_neuronx_batch_2.mar
+        url: https://torchserve.s3.amazonaws.com/mar_files/BERTSeqClassification_torchscript_neuronx_batch_2.mar
         workers:
             - 2
         batch_delay: 100
@@ -36,7 +36,7 @@ bert_neuronx_batch_2:
 bert_neuronx_batch_4:
     scripted_mode:
         benchmark_engine: "ab"
-        url: https://torchserve.pytorch.org/mar_files/BERTSeqClassification_torchscript_neuronx_batch_4.mar
+        url: https://torchserve.s3.amazonaws.com/mar_files/BERTSeqClassification_torchscript_neuronx_batch_4.mar
         workers:
             - 2
         batch_delay: 100
@@ -53,7 +53,7 @@ bert_neuronx_batch_4:
 bert_neuronx_batch_8:
     scripted_mode:
         benchmark_engine: "ab"
-        url: https://torchserve.pytorch.org/mar_files/BERTSeqClassification_torchscript_neuronx_batch_8.mar
+        url: https://torchserve.s3.amazonaws.com/mar_files/BERTSeqClassification_torchscript_neuronx_batch_8.mar
         workers:
             - 2
         batch_delay: 100

diff --git a/benchmarks/models_config/opt_6.7b_neuronx.yaml b/benchmarks/models_config/opt_6.7b_neuronx.yaml
@@ -2,7 +2,7 @@
 opt_6.7b_neuronx_batch_1:
   scripted_mode:
     benchmark_engine: "ab"
-    url: https://torchserve.pytorch.org/mar_files/opt_6.7b_neuronx_batch_1.tar.gz
+    url: https://torchserve.s3.amazonaws.com/mar_files/opt_6.7b_neuronx_batch_1.tar.gz
     workers:
       - 1
     batch_delay: 100
@@ -19,7 +19,7 @@ opt_6.7b_neuronx_batch_1:
 opt_6.7b_neuronx_batch_2:
   scripted_mode:
     benchmark_engine: "ab"
-    url: https://torchserve.pytorch.org/mar_files/opt_6.7b_neuronx_batch_2.tar.gz
+    url: https://torchserve.s3.amazonaws.com/mar_files/opt_6.7b_neuronx_batch_2.tar.gz
     workers:
       - 1
     batch_delay: 100
@@ -36,7 +36,7 @@ opt_6.7b_neuronx_batch_2:
 opt_6.7b_neuronx_batch_4:
   scripted_mode:
     benchmark_engine: "ab"
-    url: https://torchserve.pytorch.org/mar_files/opt_6.7b_neuronx_batch_4.tar.gz
+    url: https://torchserve.s3.amazonaws.com/mar_files/opt_6.7b_neuronx_batch_4.tar.gz
     workers:
       - 1
     batch_delay: 100
@@ -53,7 +53,7 @@ opt_6.7b_neuronx_batch_4:
 opt_6.7b_neuronx_batch_8:
   scripted_mode:
     benchmark_engine: "ab"
-    url: https://torchserve.pytorch.org/mar_files/opt_6.7b_neuronx_batch_8.tar.gz
+    url: https://torchserve.s3.amazonaws.com/mar_files/opt_6.7b_neuronx_batch_8.tar.gz
     workers:
       - 1
     batch_delay: 100