From 79e9455dc7b29f8e432dae58be43364329e3bf16 Mon Sep 17 00:00:00 2001 From: yuhan Date: Fri, 30 May 2025 16:26:13 +0800 Subject: [PATCH] shielded mindformers interface --- .../docs/source_en/_templates/classtemplate.rst | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/docs/mindformers/docs/source_en/_templates/classtemplate.rst b/docs/mindformers/docs/source_en/_templates/classtemplate.rst index 2fd50a2bcc..fd31c0a0d2 100644 --- a/docs/mindformers/docs/source_en/_templates/classtemplate.rst +++ b/docs/mindformers/docs/source_en/_templates/classtemplate.rst @@ -98,7 +98,7 @@ {{ fullname | underline }} .. autoclass:: {{ name }} - :exclude-members: from_dict, from_model_config, to_dict, update + :exclude-members: from_dict, from_model_config, to_dict, update, from_pretrained :members: {% elif fullname=="mindformers.generation.GenerationMixin" %} @@ -112,7 +112,7 @@ {{ fullname | underline }} .. autoclass:: {{ name }} - :exclude-members: add_flags_custom, prepare_inputs_for_generation, prepare_inputs_for_predict_layout, construct + :exclude-members: add_flags_custom, prepare_inputs_for_generation, prepare_inputs_for_predict_layout, construct, convert_map_dict, convert_name, convert_weight_dict :members: {% elif fullname=="mindformers.models.ChatGLM3Tokenizer" %} @@ -203,7 +203,7 @@ {{ fullname | underline }} .. autoclass:: {{ name }} - :exclude-members: auto_register + :exclude-members: auto_register, get_instance_type_from_cfg :members: {% elif fullname=="mindformers.Trainer" %} @@ -224,7 +224,14 @@ {{ fullname | underline }} .. autoclass:: {{ name }} - :exclude-members: epoch_begin, epoch_end, step_begin, step_end, on_train_epoch_begin, on_train_step_begin, on_train_step_end + :exclude-members: epoch_begin, epoch_end, step_begin, step_end, on_train_epoch_begin, on_train_step_begin, on_train_step_end, abnormal_global_norm_check + :members: + +{% elif fullname=="mindformers.dataset.CausalLanguageModelDataset" %} +{{ fullname | underline }} + +.. autoclass:: {{ name }} + :exclude-members: perform_token_counting, construct :members: {% elif fullname in ["mindformers.AutoModelForCausalLM", "mindformers.AutoModelForZeroShotImageClassification", "mindformers.AutoModel"] %} -- Gitee