Skip to content

Commit

Permalink
Internal change.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 699236187
  • Loading branch information
Orbax Authors committed Nov 22, 2024
1 parent 65c9068 commit 922d408
Showing 1 changed file with 0 additions and 13 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -74,19 +74,6 @@



async def _create_param_save_dir(param_info: ParamInfo):
# Directory will be unused.
path = param_info.path
if path is None:
return
# TODO(b/273803615): Note that keys with slashes ('/', generated by Haiku,
# for example) will result in the creation of nested sub-directories, rather
# than flat parameter directories like for a standard nested PyTree. This
# discrepancy, while potentially problematic, will not be addressed since we
# anticipate moving fully to OCDBT within a quarter or two.
await utils.async_makedirs(path, parents=True)


def _default_sizeof_values(values: Sequence[Any]) -> Sequence[int]:
return [sys.getsizeof(v) for v in values]

Expand Down

0 comments on commit 922d408

Please sign in to comment.