phpaiola commited on
Commit
ff83a87
1 Parent(s): b38e661

Delete xtuner_config.py

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