Update benchmark test branch

pytorch · Jun 5, 2024 · b9a79fe · b9a79fe
1 parent 2ae8286
commit b9a79fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmarks/benchmark_config_neuronx.yaml b/benchmarks/benchmark_config_neuronx.yaml
@@ -3,8 +3,8 @@
 #  - nightly: "2022.3.16"
 #  - release: "0.5.3"
 # Nightly build will be installed if "ts_version" is not specifiged
-#ts_version:
-#    branch: &ts_version "master"
+ts_version:
+    branch: &ts_version "inf2-benchmark-fix"
 
 # a list of model configure yaml files defined in benchmarks/models_config
 # or a list of model configure yaml files with full path