Views
No views yet

</s> as a stop token.| Model | Average | AGIEval | GPT4All | TruthfulQA | Bigbench |
|---|---|---|---|---|---|
| mlabonne/Gemmalpaca-2B 📄 | 38.39 | 24.48 | 51.22 | 47.02 | 30.85 |
| google/gemma-2b-it 📄 | 36.1 | 23.76 | 43.6 | 47.64 | 29.41 |
| google/gemma-2b 📄 | 34.26 | 22.7 | 43.35 | 39.96 | 31.03 |
1base_model: alpindale/gemma-2b
2model_type: GemmaForCausalLM
3tokenizer_type: GemmaTokenizer
4
5load_in_8bit: false
6load_in_4bit: true
7strict: false
8
9datasets:
10 - path: vicgalle/alpaca-gpt4
11 type: alpaca
12
13dataset_prepared_path:
14val_set_size: 0.01
15output_dir: ./out
16
17sequence_len: 2048
18sample_packing: true
19pad_to_sequence_len: true
20
21adapter: qlora
22lora_model_dir:
23lora_r: 32
24lora_alpha: 64
25lora_dropout: 0.05
26lora_target_linear: true
27
28wandb_project: axolotl
29wandb_entity:
30wandb_watch:
31wandb_name:
32wandb_log_model:
33
34gradient_accumulation_steps: 4
35micro_batch_size: 2
36num_epochs: 3
37optimizer: adamw_bnb_8bit
38lr_scheduler: cosine
39learning_rate: 0.0002
40
41train_on_inputs: false
42group_by_length: false
43bf16: auto
44fp16:
45tf32: false
46
47gradient_checkpointing: true
48early_stopping_patience:
49resume_from_checkpoint:
50local_rank:
51logging_steps: 1
52xformers_attention:
53flash_attention:
54
55warmup_steps: 10
56evals_per_epoch: 4
57eval_table_size:
58eval_table_max_new_tokens: 128
59saves_per_epoch: 1
60debug:
61deepspeed:
62weight_decay: 0.1
63fsdp:
64fsdp_config:
65special_tokens:
66 bos_token: <s>
67 eos_token: </s>
68 unk_token: <unk>