Views
No views yet


1base_model: alnrg2arg/blockchainlabs_7B_merged_test2_4
2model_type: MistralForCausalLM
3tokenizer_type: LlamaTokenizer
4is_mistral_derived_model: true
5
6load_in_8bit: false
7load_in_4bit: true
8strict: false
9
10datasets:
11 - path: NeuralNovel/Neural-Story-v1
12 type: completion
13dataset_prepared_path: last_run_prepared
14val_set_size: 0.1
15output_dir: ./qlora-out
16
17adapter: qlora
18lora_model_dir:
19
20sequence_len: 8192
21sample_packing: false
22pad_to_sequence_len: true
23
24lora_r: 32
25lora_alpha: 16
26lora_dropout: 0.05
27lora_target_linear: true
28lora_fan_in_fan_out:
29lora_target_modules:
30 - gate_proj
31 - down_proj
32 - up_proj
33 - q_proj
34 - v_proj
35 - k_proj
36 - o_proj
37
38wandb_project:
39wandb_entity:
40wandb_watch:
41wandb_name:
42wandb_log_model:
43
44gradient_accumulation_steps: 4
45micro_batch_size: 2
46num_epochs: 1
47optimizer: adamw_bnb_8bit
48lr_scheduler: cosine
49learning_rate: 0.0002
50
51train_on_inputs: false
52group_by_length: false
53bf16: true
54fp16: false
55tf32: false
56
57gradient_checkpointing: true
58early_stopping_patience:
59resume_from_checkpoint:
60local_rank:
61logging_steps: 1
62xformers_attention:
63flash_attention: true
64
65loss_watchdog_threshold: 5.0
66loss_watchdog_patience: 3
67
68warmup_steps: 10
69evals_per_epoch: 4
70eval_table_size:
71eval_table_max_new_tokens: 128
72saves_per_epoch: 1
73debug:
74deepspeed:
75weight_decay: 0.0
76fsdp:
77fsdp_config:
78special_tokens:
79 bos_token: "<s>"
80 eos_token: "</s>"
81 unk_token: "<unk>"
820.3.0bitsandbytes quantization config was used during training:| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 2.3251 | 0.06 | 1 | 2.8409 |
| 2.5318 | 0.25 | 4 | 2.7634 |
| 1.7316 | 0.51 | 8 | 2.3662 |
| 1.5196 | 0.76 | 12 | 2.1411 |
| Metric | Value |
|---|---|
| Avg. | 74.21 |
| AI2 Reasoning Challenge (25-Shot) | 72.27 |
| HellaSwag (10-Shot) | 86.59 |
| MMLU (5-Shot) | 64.09 |
| TruthfulQA (0-shot) | 69.84 |
| Winogrande (5-shot) | 83.35 |
| GSM8k (5-shot) | 69.14 |