Basic component Args: manager (object): An object for unified management of components
(manager)
| 31 | |
| 32 | |
| 33 | def build(manager): |
| 34 | """ |
| 35 | Basic component |
| 36 | Args: |
| 37 | manager (object): An object for unified management of components |
| 38 | """ |
| 39 | |
| 40 | default_basic_language = config.get_default_user_dict("basic", "language") |
| 41 | default_basic_model_name = config.get_default_user_dict("basic", "model_name") |
| 42 | default_basic_model_source = config.get_default_user_dict("basic", "model_source") |
| 43 | default_basic_model_name_or_path = config.get_default_user_dict("basic", "model_name_or_path") |
| 44 | default_basic_fine_tuning = config.get_default_user_dict("basic", "fine_tuning") |
| 45 | default_basic_output_dir = config.get_default_user_dict("basic", "output_dir") |
| 46 | default_basic_amp_master_grad = str(config.get_default_user_dict("basic", "amp_master_grad")) |
| 47 | default_basic_compute_type = config.get_default_user_dict("basic", "compute_type") |
| 48 | default_basic_tensor_parallel_degree = config.get_default_user_dict("basic", "tensor_parallel_degree") |
| 49 | default_basic_pipeline_parallel_degree = config.get_default_user_dict("basic", "pipeline_parallel_degree") |
| 50 | default_basic_sharding_parallel_degree = config.get_default_user_dict("basic", "sharding_parallel_degree") |
| 51 | default_basic_pipeline_parallel_config = config.get_default_user_dict("basic", "pipeline_parallel_config") |
| 52 | default_basic_pp_seg_method = config.get_default_user_dict("basic", "pp_seg_method") |
| 53 | default_basic_use_sp_callback = str(config.get_default_user_dict("basic", "use_sp_callback")) |
| 54 | default_basic_sharding = config.get_default_user_dict("basic", "sharding") |
| 55 | default_basic_moe_group = config.get_default_user_dict("basic", "moe_group") |
| 56 | default_basic_disable_ckpt_quant = str(config.get_default_user_dict("basic", "disable_ckpt_quant")) |
| 57 | default_basic_lora_rank = config.get_default_user_dict("basic", "lora_rank") |
| 58 | default_basic_lora_alpha = config.get_default_user_dict("basic", "lora_alpha") |
| 59 | default_basic_lora_plus_scale = config.get_default_user_dict("basic", "lora_plus_scale") |
| 60 | default_basic_rslora = str(config.get_default_user_dict("basic", "rslora")) |
| 61 | |
| 62 | with gr.Row(elem_classes="basic-info-row"): |
| 63 | with gr.Row(scale=1): |
| 64 | language = gr.Dropdown( |
| 65 | choices=config.get_choices_kwargs("language"), |
| 66 | value=default_basic_language, |
| 67 | ) |
| 68 | |
| 69 | with gr.Row(scale=2): |
| 70 | model_name = gr.Dropdown( |
| 71 | choices=config.get_choices_kwargs("model_name"), |
| 72 | value=default_basic_model_name, |
| 73 | interactive=False, |
| 74 | ) |
| 75 | with gr.Row(scale=1): |
| 76 | model_source = gr.Dropdown( |
| 77 | choices=config.get_choices_kwargs("model_source_ernie"), |
| 78 | value=default_basic_model_source, |
| 79 | interactive=False, |
| 80 | ) |
| 81 | |
| 82 | with gr.Row(scale=3): |
| 83 | model_name_or_path = gr.Textbox( |
| 84 | value=default_basic_model_name_or_path, |
| 85 | ) |
| 86 | |
| 87 | with gr.Row(elem_classes="basic-info-row"): |
| 88 | |
| 89 | output_dir = gr.Textbox( |
| 90 | visible=False, |
nothing calls this directly
no test coverage detected