Merge branch 'master' into oncall

GoogleCloudPlatform · Nov 25, 2024 · 5178c83 · 5178c83
2 parents a4dc2c8 + 2a3a8e8
commit 5178c83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/solutions_team/solutionsteam_tf_nightly_supported.py b/dags/solutions_team/solutionsteam_tf_nightly_supported.py
@@ -112,7 +112,7 @@
       runtime_version=RuntimeVersion.TPU_VM_TF_NIGHTLY_POD.value,
   )
 
-  embedding_dim = 32
+  embedding_dim = 16
   tf_dlrm_v5p_8 = tf_config.get_tf_dlrm_config(
       project_name=Project.TPU_PROD_ENV_AUTOMATED.value,
       tpu_version=TpuVersion.V5P,