File size: 1,321 Bytes
5ab514a 29ac941 5ab514a 29ac941 5ab514a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
{
"model_type": "llama",
"alpha_pattern": {},
"auto_mapping": null,
"base_model_name_or_path": "unsloth/llama-3-8b-bnb-4bit",
"bias": "none",
"fan_in_fan_out": false,
"inference_mode": true,
"init_lora_weights": true,
"layer_replication": null,
"layers_pattern": null,
"layers_to_transform": null,
"loftq_config": {},
"lora_alpha": 16,
"lora_dropout": 0,
"megatron_config": null,
"megatron_core": "megatron.core",
"modules_to_save": [
"lm_head",
"embed_tokens"
],
"peft_type": "LORA",
"r": 16,
"rank_pattern": {},
"revision": "unsloth",
"target_modules": [
"k_proj",
"gate_proj",
"q_proj",
"up_proj",
"o_proj",
"down_proj",
"v_proj"
],
"task_type": "CAUSAL_LM",
"use_dora": false,
"use_rslora": false,
"hidden_size": 4096,
"num_attention_heads": 32,
"num_hidden_layers": 24,
"intermediate_size": 16384,
"max_position_embeddings": 512,
"vocab_size": 32000,
"layer_norm_eps": 1e-5,
"initializer_range": 0.02,
"train_batch_size": 2,
"gradient_accumulation_steps": 4,
"warmup_steps": 5,
"max_steps": 60,
"learning_rate": 0.0002,
"fp16": true,
"bf16": false,
"logging_steps": 1,
"optim": "adamw_8bit",
"weight_decay": 0.01,
"lr_scheduler_type": "linear",
"seed": 3407,
"output_dir": "outputs"
} |