Skip to content

Commit

Permalink
Update MaxText/prefix_cache.py docstring
Browse files Browse the repository at this point in the history
Co-authored-by: RupengLiu <[email protected]>
  • Loading branch information
yuyanpeng-google and liurupeng authored Feb 26, 2025
1 parent f42d4e0 commit a8f1544
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion MaxText/prefix_cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ def prefix_size_bytes(self) -> int:
return self._prefix_size_bytes

def clone(self) -> "Value":
"""Clone to prevent use the same jax array."""
"""Clone to prevent using the same jax array."""
copied_prefix = tree_copy(self._prefix)
return Value(
prefix=copied_prefix,
Expand Down

0 comments on commit a8f1544

Please sign in to comment.