| { |
| "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" |
| } |
