From 2fce2487f6eabb039cba3015354af4298bbdb104 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 27 Dec 2023 00:06:38 -0500 Subject: [PATCH] [zeta.utils][cleanup] --- file_list.txt | 35 ++++++++++++++++++++++++++ mkdocs.yml | 70 +++++++++++++++++++++++++-------------------------- 2 files changed, 70 insertions(+), 35 deletions(-) create mode 100644 file_list.txt diff --git a/file_list.txt b/file_list.txt new file mode 100644 index 00000000..865c4391 --- /dev/null +++ b/file_list.txt @@ -0,0 +1,35 @@ +- cast_tuple: "zeta/utils/cast_tuple.md" +- group_by_key_prefix: "zeta/utils/group_by_key_prefix.md" +- eval_decorator: "zeta/utils/eval_decorator.md" +- print_cuda_memory_usage: "zeta/utils/print_cuda_memory_usage.md" +- once: "zeta/utils/once.md" +- default: "zeta/utils/default.md" +- gumbel_noise: "zeta/utils/gumbel_noise.md" +- pad_at_dim: "zeta/utils/pad_at_dim.md" +- init_zero_: "zeta/utils/init_zero_.md" +- top_p: "zeta/utils/top_p.md" +- cast_if_src_dtype: "zeta/utils/cast_if_src_dtype.md" +- disable_warnings_and_logs: "zeta/utils/disable_warnings_and_logs.md" +- save_load_wrapper: "zeta/utils/save_load_wrapper.md" +- get_sinusoid_encoding_table: "zeta/utils/get_sinusoid_encoding_table.md" +- main: "zeta/utils/main.md" +- string_begins_with: "zeta/utils/string_begins_with.md" +- gif_to_tensor: "zeta/utils/gif_to_tensor.md" +- l2norm: "zeta/utils/l2norm.md" +- save_load: "zeta/utils/save_load.md" +- log: "zeta/utils/log.md" +- module_device: "zeta/utils/module_device.md" +- print_num_params: "zeta/utils/print_num_params.md" +- top_a: "zeta/utils/top_a.md" +- interpolate_pos_encoding_2d: "zeta/utils/interpolate_pos_encoding_2d.md" +- exists: "zeta/utils/exists.md" +- cosine_beta_schedule: "zeta/utils/cosine_beta_schedule.md" +- track_cuda_memory: "zeta/utils/track_cuda_memory.md" +- maybe: "zeta/utils/maybe.md" +- save_memory_snapshot: "zeta/utils/save_memory_snapshot.md" +- top_k: "zeta/utils/top_k.md" +- print_main: "zeta/utils/print_main.md" +- pick_and_pop: "zeta/utils/pick_and_pop.md" +- track_cuda_memory_usage: "zeta/utils/track_cuda_memory_usage.md" +- group_dict_by_key: "zeta/utils/group_dict_by_key.md" +- video_tensor_to_gift: "zeta/utils/video_tensor_to_gift.md" diff --git a/mkdocs.yml b/mkdocs.yml index 6d716b7b..09f6e334 100644 --- a/mkdocs.yml +++ b/mkdocs.yml @@ -163,41 +163,41 @@ nav: - SentencePieceTokenizer: "zeta/tokenizers/sentencepiece.md" - TokenMonster: "zeta/tokenizers/token_monster.md" - zeta.utils: - - cast_tuple: "cast_tuple.md" - - group_by_key_prefix: "group_by_key_prefix.md" - - eval_decorator: "eval_decorator.md" - - print_cuda_memory_usage: "print_cuda_memory_usage.md" - - once: "once.md" - - default: "default.md" - - gumbel_noise: "gumbel_noise.md" - - pad_at_dim: "pad_at_dim.md" - - init_zero_: "init_zero_.md" - - top_p: "top_p.md" - - cast_if_src_dtype: "cast_if_src_dtype.md" - - disable_warnings_and_logs: "disable_warnings_and_logs.md" - - save_load_wrapper: "save_load_wrapper.md" - - get_sinusoid_encoding_table: "get_sinusoid_encoding_table.md" - - main: "main.md" - - string_begins_with: "string_begins_with.md" - - gif_to_tensor: "gif_to_tensor.md" - - l2norm: "l2norm.md" - - save_load: "save_load.md" - - log: "log.md" - - module_device: "module_device.md" - - print_num_params: "print_num_params.md" - - top_a: "top_a.md" - - interpolate_pos_encoding_2d: "interpolate_pos_encoding_2d.md" - - exists: "exists.md" - - cosine_beta_schedule: "cosine_beta_schedule.md" - - track_cuda_memory: "track_cuda_memory.md" - - maybe: "maybe.md" - - save_memory_snapshot: "save_memory_snapshot.md" - - top_k: "top_k.md" - - print_main: "print_main.md" - - pick_and_pop: "pick_and_pop.md" - - track_cuda_memory_usage: "track_cuda_memory_usage.md" - - group_dict_by_key: "group_dict_by_key.md" - - video_tensor_to_gift: "video_tensor_to_gift.md" + - cast_tuple: "zeta/utils/cast_tuple.md" + - group_by_key_prefix: "zeta/utils/group_by_key_prefix.md" + - eval_decorator: "zeta/utils/eval_decorator.md" + - print_cuda_memory_usage: "zeta/utils/print_cuda_memory_usage.md" + - once: "zeta/utils/once.md" + - default: "zeta/utils/default.md" + - gumbel_noise: "zeta/utils/gumbel_noise.md" + - pad_at_dim: "zeta/utils/pad_at_dim.md" + - init_zero_: "zeta/utils/init_zero_.md" + - top_p: "zeta/utils/top_p.md" + - cast_if_src_dtype: "zeta/utils/cast_if_src_dtype.md" + - disable_warnings_and_logs: "zeta/utils/disable_warnings_and_logs.md" + - save_load_wrapper: "zeta/utils/save_load_wrapper.md" + - get_sinusoid_encoding_table: "zeta/utils/get_sinusoid_encoding_table.md" + - main: "zeta/utils/main.md" + - string_begins_with: "zeta/utils/string_begins_with.md" + - gif_to_tensor: "zeta/utils/gif_to_tensor.md" + - l2norm: "zeta/utils/l2norm.md" + - save_load: "zeta/utils/save_load.md" + - log: "zeta/utils/log.md" + - module_device: "zeta/utils/module_device.md" + - print_num_params: "zeta/utils/print_num_params.md" + - top_a: "zeta/utils/top_a.md" + - interpolate_pos_encoding_2d: "zeta/utils/interpolate_pos_encoding_2d.md" + - exists: "zeta/utils/exists.md" + - cosine_beta_schedule: "zeta/utils/cosine_beta_schedule.md" + - track_cuda_memory: "zeta/utils/track_cuda_memory.md" + - maybe: "zeta/utils/maybe.md" + - save_memory_snapshot: "zeta/utils/save_memory_snapshot.md" + - top_k: "zeta/utils/top_k.md" + - print_main: "zeta/utils/print_main.md" + - pick_and_pop: "zeta/utils/pick_and_pop.md" + - track_cuda_memory_usage: "zeta/utils/track_cuda_memory_usage.md" + - group_dict_by_key: "zeta/utils/group_dict_by_key.md" + - video_tensor_to_gift: "zeta/utils/video_tensor_to_gift.md" - zeta.ops: - main: "zeta/ops/main.md" - softmaxes: "zeta/ops/softmaxes.md"