[zeta.models][fix]

file_list.txt

@@ -1,35 +1,10 @@
-- 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"
+- vit: "zeta/modelsvit.md"
+- gpt4multimodal: "zeta/modelsgpt4multimodal.md"
+- maxvit: "zeta/modelsmaxvit.md"
+- llama2: "zeta/modelsllama2.md"
+- gpt4: "zeta/modelsgpt4.md"
+- andromeda: "zeta/modelsandromeda.md"
+- basemodel: "zeta/modelsbasemodel.md"
+- palme: "zeta/modelspalme.md"
+- megavit: "zeta/modelsmegavit.md"
+- navit: "zeta/modelsnavit.md"

mkdocs.yml

@@ -209,16 +209,16 @@ nav:
           - ParallelWrapper: "zeta/training/parallel_wrapper.md"
           - train: "zeta/training/train.md"
       - zeta.models:
-          - vit: "vit.md"
-          - gpt4multimodal: "gpt4multimodal.md"
-          - maxvit: "maxvit.md"
-          - llama2: "llama2.md"
-          - gpt4: "gpt4.md"
-          - andromeda: "andromeda.md"
-          - basemodel: "basemodel.md"
-          - palme: "palme.md"
-          - megavit: "megavit.md"
-          - navit: "navit.md"
+        - vit: "zeta/modelsvit.md"
+        - gpt4multimodal: "zeta/modelsgpt4multimodal.md"
+        - maxvit: "zeta/modelsmaxvit.md"
+        - llama2: "zeta/modelsllama2.md"
+        - gpt4: "zeta/modelsgpt4.md"
+        - andromeda: "zeta/modelsandromeda.md"
+        - basemodel: "zeta/modelsbasemodel.md"
+        - palme: "zeta/modelspalme.md"
+        - megavit: "zeta/modelsmegavit.md"
+        - navit: "zeta/modelsnavit.md"
       - zeta.quant:
           - QUIK: "zeta/quant/quik.md"
           - BitLinear: "zeta/quant/bitlinear.md"

scripts/auto_tests_docs/mkdocs_handler.py

@@ -22,8 +22,8 @@ def generate_file_list(directory, output_file):
                     # Remove the file extension
                     file_name, _ = os.path.splitext(file)
                     # Write the file name and path to the output file
-                    f.write(f'- {file_name}: "{file_path}"\n')
+                    f.write(f'- {file_name}: "{directory}{file_path}"\n')
 
 
 # Use the function to generate the file list
-generate_file_list("docs/zeta/utils", "file_list.txt")
+generate_file_list("docs/zeta/models", "file_list.txt")