Shortening model names to be under 40 chars.

AI-Hypercomputer · Dec 16, 2024 · cd1999e · cd1999e
1 parent 77c8c04
commit cd1999e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/benchmarks/benchmark_runner.py b/benchmarks/benchmark_runner.py
@@ -89,8 +89,8 @@ def add_shared_arguments(custom_parser: argparse.ArgumentParser):
           'llama2_70b_4096_real_data',
           'llama2_70b_4096_pw_long_run',
           'llama2_70b_4096_real_data_pw_long_run',
-          'llama2_70b_4096_sc_synthetic_pw_lr',
-          'llama2_70b_4096_sc_synthetic',
+          'llama2_70b_4096_synthetic_pw_lr',
+          'llama2_70b_4096_synthetic',
           'llama3_70b_8192',
           'llama3_1_405b_8192_fsdp_dcn',
           'mixtral_8x7b_dropped',
@@ -108,8 +108,8 @@ def add_shared_arguments(custom_parser: argparse.ArgumentParser):
           'llama2_70b_4096_real_data '
           'llama2_70b_4096_pw_long_run '
           'llama2_70b_4096_real_data_pw_long_run '
-          'llama2_70b_4096_sc_synthetic_pw_lr '
-          'llama2_70b_4096_sc_synthetic '
+          'llama2_70b_4096_synthetic_pw_lr '
+          'llama2_70b_4096_synthetic '
           'llama3_1_405b_8192_fsdp_dcn '
           'mixtral_8x7b_dropped '
           'mixtral_8x7b_dropped_int8 '

diff --git a/benchmarks/maxtext_trillium_model_configs.py b/benchmarks/maxtext_trillium_model_configs.py
@@ -358,8 +358,8 @@ class MaxTextModel:
         + xla_flags_library.CF_FOR_ALL_GATHER
     ),
 )
-llama2_70b_4096_sc_synthetic = MaxTextModel(
-    model_name="llama2_70b_4096_sc_synthetic",
+llama2_70b_4096_synthetic = MaxTextModel(
+    model_name="llama2_70b_4096_synthetic",
     model_type="llama2-70b",
     tuning_params={
         "per_device_batch_size": 2,
@@ -385,8 +385,8 @@ class MaxTextModel:
     ),
 )
 
-llama2_70b_4096_sc_synthetic_pw_lr = MaxTextModel(
-    model_name="llama2_70b_4096_sc_synthetic_pw_lr",
+llama2_70b_4096_synthetic_pw_lr = MaxTextModel(
+    model_name="llama2_70b_4096_synthetic_pw_lr",
     model_type="llama2-70b",
     tuning_params={
         "per_device_batch_size": 2,
@@ -762,8 +762,8 @@ class MaxTextModel:
     llama2_70b_4096_real_data_pw_long_run,
     llama3_8b_8192,  # Not Optimizied yet
     llama3_70b_8192,  # Not Optimizied yet
-    llama2_70b_4096_sc_synthetic_pw_lr,
-    llama2_70b_4096_sc_synthetic,
+    llama2_70b_4096_synthetic_pw_lr,
+    llama2_70b_4096_synthetic,
     llama3_1_405b_8192_fsdp_dcn,
     llama3_1_70b_129024,
     mixtral_8x7b_dropped,