phpaiola commited on
Commit
66e1132
1 Parent(s): 184a2f1

Delete xtuner_config.py

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