minor fix

google · Jun 25, 2024 · 9c0d855 · 9c0d855
1 parent 5e85ec3
commit 9c0d855
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/MaxText/maxengine.py b/MaxText/maxengine.py
@@ -323,8 +323,6 @@ def copy(path, partial_cache, full_cache, annotations):
  inserted_generated_tokens = jax.lax.dynamic_update_index_in_dim(
  decode_state["generated_tokens"], unboxed_prefix["generated_tokens"], slot, 0
  )
- logging.info("decode_state keys: {}".format(decode_state.keys()))
- logging.info("unboxed_prefix keys: {}".format(unboxed_prefix.keys()))
  inserted_tokens = jax.lax.dynamic_update_index_in_dim(
  decode_state["tokens"], unboxed_prefix["tokens"], slot, 0
  )