phpaiola commited on
Commit
267d6f1
·
verified ·
1 Parent(s): abb851b

Delete xtuner_config.py

Browse files
Files changed (1) hide show
  1. xtuner_config.py +0 -189
xtuner_config.py DELETED
@@ -1,189 +0,0 @@
1
- SYSTEM = 'xtuner.utils.SYSTEM_TEMPLATE.alpaca'
2
- accumulative_counts = 16
3
- alpaca_en = dict(
4
- dataset=dict(
5
- data_files='/petrobr/parceirosbr/radiar/llama_test/ultracabrita3.json',
6
- path='json',
7
- type='datasets.load_dataset'),
8
- dataset_map_fn='xtuner.dataset.map_fns.ultracabrita_map_fn',
9
- max_length=4096,
10
- pack_to_max_length=True,
11
- remove_unused_columns=True,
12
- shuffle_before_pack=True,
13
- template_map_fn=dict(
14
- template='xtuner.utils.PROMPT_TEMPLATE.internlm2_chat',
15
- type='xtuner.dataset.map_fns.template_map_fn_factory'),
16
- tokenizer=dict(
17
- padding_side='right',
18
- pretrained_model_name_or_path=
19
- '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d',
20
- trust_remote_code=True,
21
- type='transformers.AutoTokenizer.from_pretrained'),
22
- type='xtuner.dataset.process_hf_dataset',
23
- use_varlen_attn=False)
24
- alpaca_en_path = '/petrobr/parceirosbr/radiar/llama_test/ultracabrita3.json'
25
- batch_size = 2
26
- betas = (
27
- 0.9,
28
- 0.999,
29
- )
30
- custom_hooks = [
31
- dict(
32
- tokenizer=dict(
33
- padding_side='right',
34
- pretrained_model_name_or_path=
35
- '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d',
36
- trust_remote_code=True,
37
- type='transformers.AutoTokenizer.from_pretrained'),
38
- type='xtuner.engine.hooks.DatasetInfoHook'),
39
- dict(
40
- evaluation_inputs=[
41
- 'O que é um bode?',
42
- 'Qual a capital da França?',
43
- 'Você pode me explicar o Teorema de Pitágoras com um exemplo, por favor?',
44
- 'Olá, tudo bem? Estou procurando livros de ficção científica para ler, você teria sugestões para mim?',
45
- ],
46
- every_n_iters=500,
47
- prompt_template='xtuner.utils.PROMPT_TEMPLATE.internlm2_chat',
48
- system='xtuner.utils.SYSTEM_TEMPLATE.alpaca',
49
- tokenizer=dict(
50
- padding_side='right',
51
- pretrained_model_name_or_path=
52
- '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d',
53
- trust_remote_code=True,
54
- type='transformers.AutoTokenizer.from_pretrained'),
55
- type='xtuner.engine.hooks.EvaluateChatHook'),
56
- ]
57
- dataloader_num_workers = 0
58
- default_hooks = dict(
59
- checkpoint=dict(
60
- by_epoch=False,
61
- interval=500,
62
- max_keep_ckpts=2,
63
- type='mmengine.hooks.CheckpointHook'),
64
- logger=dict(
65
- interval=10,
66
- log_metric_by_epoch=False,
67
- type='mmengine.hooks.LoggerHook'),
68
- param_scheduler=dict(type='mmengine.hooks.ParamSchedulerHook'),
69
- sampler_seed=dict(type='mmengine.hooks.DistSamplerSeedHook'),
70
- timer=dict(type='mmengine.hooks.IterTimerHook'))
71
- env_cfg = dict(
72
- cudnn_benchmark=False,
73
- dist_cfg=dict(backend='nccl'),
74
- mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0))
75
- evaluation_freq = 500
76
- evaluation_inputs = [
77
- 'O que é um bode?',
78
- 'Qual a capital da França?',
79
- 'Você pode me explicar o Teorema de Pitágoras com um exemplo, por favor?',
80
- 'Olá, tudo bem? Estou procurando livros de ficção científica para ler, você teria sugestões para mim?',
81
- ]
82
- launcher = 'pytorch'
83
- load_from = None
84
- log_level = 'INFO'
85
- log_processor = dict(by_epoch=False)
86
- lr = 2e-05
87
- max_epochs = 1
88
- max_length = 4096
89
- max_norm = 1
90
- model = dict(
91
- llm=dict(
92
- pretrained_model_name_or_path=
93
- '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d',
94
- trust_remote_code=True,
95
- type='transformers.AutoModelForCausalLM.from_pretrained'),
96
- type='xtuner.model.SupervisedFinetune',
97
- use_varlen_attn=False)
98
- optim_type = 'torch.optim.AdamW'
99
- optim_wrapper = dict(
100
- optimizer=dict(
101
- betas=(
102
- 0.9,
103
- 0.999,
104
- ),
105
- lr=2e-05,
106
- type='torch.optim.AdamW',
107
- weight_decay=0),
108
- type='DeepSpeedOptimWrapper')
109
- pack_to_max_length = True
110
- param_scheduler = [
111
- dict(
112
- begin=0,
113
- by_epoch=True,
114
- convert_to_iter_based=True,
115
- end=0.03,
116
- start_factor=1e-05,
117
- type='mmengine.optim.LinearLR'),
118
- dict(
119
- begin=0.03,
120
- by_epoch=True,
121
- convert_to_iter_based=True,
122
- end=1,
123
- eta_min=0.0,
124
- type='mmengine.optim.CosineAnnealingLR'),
125
- ]
126
- pretrained_model_name_or_path = '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d'
127
- prompt_template = 'xtuner.utils.PROMPT_TEMPLATE.internlm2_chat'
128
- randomness = dict(deterministic=False, seed=None)
129
- resume = False
130
- runner_type = 'FlexibleRunner'
131
- save_steps = 500
132
- save_total_limit = 2
133
- strategy = dict(
134
- config=dict(
135
- bf16=dict(enabled=False),
136
- fp16=dict(enabled=True, initial_scale_power=16),
137
- gradient_accumulation_steps='auto',
138
- gradient_clipping='auto',
139
- train_micro_batch_size_per_gpu='auto',
140
- zero_allow_untested_optimizer=True,
141
- zero_force_ds_cpu_optimizer=False,
142
- zero_optimization=dict(overlap_comm=True, stage=2)),
143
- exclude_frozen_parameters=True,
144
- gradient_accumulation_steps=16,
145
- gradient_clipping=1,
146
- sequence_parallel_size=1,
147
- train_micro_batch_size_per_gpu=2,
148
- type='xtuner.engine.DeepSpeedStrategy')
149
- tokenizer = dict(
150
- padding_side='right',
151
- pretrained_model_name_or_path=
152
- '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d',
153
- trust_remote_code=True,
154
- type='transformers.AutoTokenizer.from_pretrained')
155
- train_cfg = dict(max_epochs=1, type='xtuner.engine.runner.TrainLoop')
156
- train_dataloader = dict(
157
- batch_size=2,
158
- collate_fn=dict(
159
- type='xtuner.dataset.collate_fns.default_collate_fn',
160
- use_varlen_attn=False),
161
- dataset=dict(
162
- dataset=dict(
163
- data_files=
164
- '/petrobr/parceirosbr/radiar/llama_test/ultracabrita3.json',
165
- path='json',
166
- type='datasets.load_dataset'),
167
- dataset_map_fn='xtuner.dataset.map_fns.ultracabrita_map_fn',
168
- max_length=4096,
169
- pack_to_max_length=True,
170
- remove_unused_columns=True,
171
- shuffle_before_pack=True,
172
- template_map_fn=dict(
173
- template='xtuner.utils.PROMPT_TEMPLATE.internlm2_chat',
174
- type='xtuner.dataset.map_fns.template_map_fn_factory'),
175
- tokenizer=dict(
176
- padding_side='right',
177
- pretrained_model_name_or_path=
178
- '/petrobr/parceirosbr/home/luis.afonso/.cache/huggingface/hub/models--internlm--internlm2-chat-1_8b/snapshots/ecccbb5c87079ad84e5788baa55dd6e21a9c614d',
179
- trust_remote_code=True,
180
- type='transformers.AutoTokenizer.from_pretrained'),
181
- type='xtuner.dataset.process_hf_dataset',
182
- use_varlen_attn=False),
183
- num_workers=0,
184
- sampler=dict(shuffle=True, type='mmengine.dataset.DefaultSampler'))
185
- use_varlen_attn = False
186
- visualizer = None
187
- warmup_ratio = 0.03
188
- weight_decay = 0
189
- work_dir = './work_dirs/internlm2_chat_1_8b_full_ultracabrita'