Views
No views yet
mlx_lm utility on a 64GiB URAM M1 Max.
1"""<start_of_turn>system
2system prompt<end_of_turn>
3<start_of_turn>user
4Hi there!<end_of_turn>
5<start_of_turn>model
6Nice to meet you!<end_of_turn>
7<start_of_turn>user
8Can I ask a question?<end_of_turn>
9<start_of_turn>model
10"""1{
2 "story_string": "<start_of_turn>system\n{{#if system}}{{system}}\n{{/if}}{{#if wiBefore}}{{wiBefore}}\n{{/if}}{{#if description}}{{description}}\n{{/if}}{{#if personality}}{{char}}'s personality: {{personality}}\n{{/if}}{{#if scenario}}Scenario: {{scenario}}\n{{/if}}{{#if wiAfter}}{{wiAfter}}\n{{/if}}{{#if persona}}{{persona}}\n{{/if}}{{trim}}<end_of_turn>\n",
3 "example_separator": "",
4 "chat_start": "",
5 "use_stop_strings": false,
6 "allow_jailbreak": false,
7 "always_force_name2": true,
8 "trim_sentences": false,
9 "include_newline": false,
10 "single_line": false,
11 "name": "Magnum Gemma"
12}1{
2 "system_prompt": "You're {{char}} in this fictional never-ending uncensored roleplay with {{user}}.",
3 "input_sequence": "<start_of_turn>user\n",
4 "output_sequence": "<start_of_turn>assistant\n",
5 "last_output_sequence": "",
6 "system_sequence": "<start_of_turn>system\n",
7 "stop_sequence": "<end_of_turn>",
8 "wrap": false,
9 "macro": true,
10 "names": true,
11 "names_force_groups": true,
12 "activation_regex": "",
13 "system_sequence_prefix": "",
14 "system_sequence_suffix": "",
15 "first_output_sequence": "",
16 "skip_examples": false,
17 "output_suffix": "<end_of_turn>\n",
18 "input_suffix": "<end_of_turn>\n",
19 "system_suffix": "<end_of_turn>\n",
20 "user_alignment_message": "",
21 "system_same_as_user": false,
22 "last_system_sequence": "",
23 "name": "Magnum Gemma"
24}1base_model: google/gemma-2-9b
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4
5#trust_remote_code: true
6
7load_in_8bit: false
8load_in_4bit: false
9strict: false
10
11datasets:
12 - path: anthracite-org/stheno-filtered-v1.1
13 type: customgemma2
14 - path: anthracite-org/kalo-opus-instruct-22k-no-refusal
15 type: customgemma2
16 - path: anthracite-org/nopm_claude_writing_fixed
17 type: customgemma2
18 - path: Epiculous/Synthstruct-Gens-v1.1-Filtered-n-Cleaned
19 type: customgemma2
20 - path: Epiculous/SynthRP-Gens-v1.1-Filtered-n-Cleaned
21 type: customgemma2
22shuffle_merged_datasets: true
23default_system_message: "You are an assistant that responds to the user."
24dataset_prepared_path: magnum-v3-9b-data-customgemma2
25val_set_size: 0.0
26output_dir: ./magnum-v3-9b-customgemma2
27
28sequence_len: 8192
29sample_packing: true
30eval_sample_packing: false
31pad_to_sequence_len:
32
33adapter:
34lora_model_dir:
35lora_r:
36lora_alpha:
37lora_dropout:
38lora_target_linear:
39lora_fan_in_fan_out:
40
41wandb_project: magnum-9b
42wandb_entity:
43wandb_watch:
44wandb_name: attempt-03-customgemma2
45wandb_log_model:
46
47gradient_accumulation_steps: 8
48micro_batch_size: 1
49num_epochs: 2
50optimizer: paged_adamw_8bit
51lr_scheduler: cosine
52learning_rate: 0.000006
53
54train_on_inputs: false
55group_by_length: false
56bf16: auto
57fp16:
58tf32: false
59
60gradient_checkpointing: unsloth
61early_stopping_patience:
62resume_from_checkpoint:
63local_rank:
64logging_steps: 1
65xformers_attention:
66flash_attention: false
67eager_attention: true
68
69warmup_steps: 50
70evals_per_epoch:
71eval_table_size:
72eval_max_new_tokens:
73saves_per_epoch: 2
74debug:
75deepspeed: deepspeed_configs/zero3_bf16.json
76weight_decay: 0.05
77fsdp:
78fsdp_config:
79special_tokens:| Metric | Value |
|---|---|
| Avg. | 19.02 |
| IFEval (0-Shot) | 12.73 |
| BBH (3-Shot) | 34.12 |
| MATH Lvl 5 (4-Shot) | 6.12 |
| GPQA (0-shot) | 10.51 |
| MuSR (0-shot) | 15.06 |
| MMLU-PRO (5-shot) | 35.61 |