Click to expand full config
Axolotl version: 0.8.0.dev0
1adapter: lora
2base_model: mistralai/Mistral-7B-Instruct-v0.3
3model_type: MistralForCausalLM
4tokenizer_type: AutoTokenizer
5bf16: true
6dataset_processes: 32
7datasets:
8 - path: bytess/zrah-personal-ai
9 type: alpaca
10gradient_accumulation_steps: 4
11gradient_checkpointing: false
12learning_rate: 0.0002
13lora_alpha: 32
14lora_dropout: 0.05
15lora_r: 16 # Or try 8 for smaller size later
16lora_target_modules:
17 - q_proj
18 - v_proj
19 - k_proj
20 - o_proj
21 - gate_proj
22 - down_proj
23 - up_proj
24loraplus_lr_embedding: 1.0e-06
25lr_scheduler: cosine
26max_prompt_len: 512
27micro_batch_size: 4 # Increase from 2 if GPU allows
28num_epochs: 3
29optimizer: adamw_torch
30output_dir: ./outputs/zrah_model
31pretrain_multipack_attn: true
32sample_packing_bin_size: 200
33sample_packing_group_size: 100000
34save_only_model: true
35save_safetensors: true
36sequence_len: 2048
37shuffle_merged_datasets: true
38train_on_inputs: false
39trl:
40 use_vllm: false
41val_set_size: 0.0
42weight_decay: 0.0
Zrah Model 1.0
This model is a fine-tuned version of mistralai/Mistral-7B-Instruct-v0.3, trained using Axolotl on a RunPod instance with H100 PCIe GPU. The training was conducted for personal/self-aligned instruction tuning using a custom dataset: bytess/zrah-personal-ai.
⚠️ This model is intended only for demonstration and trial purposes as part of a personal portfolio. Use is allowed, but modification, fine-tuning, or redistribution is prohibited due to the private nature of the data.
Model Description
This model uses LoRA adapters to fine-tune Mistral-7B on a dataset of personal prompts and completions designed to align the model to the author's style, tone, and preferences. The model is intended for self-use in chat, writing assistance, and experimentation with instruction-following behavior.
Intended Uses & Limitations
Intended Uses:
- Personal assistant/chatbot
- Writing/code helper tailored to author's style
- Experimentation with LoRA fine-tuning and self-alignment
Limitations:
- Trained only on a small personal dataset; generalization to broader tasks is limited
- Not evaluated on standardized benchmarks
- May reflect idiosyncratic styles or biases present in the training data
Training and Evaluation Data
The model was trained on the bytess/zrah-personal-ai dataset. This dataset is structured in Alpaca-style (instruction, input, output) and consists of hand-curated personal instructions and completions.
- Number of examples: 130 rows of data
- Validation set: None used (
val_set_size: 0.0)
- Data shuffled and packed for efficiency using Axolotl's sample packing
Training Procedure
Hyperparameters
- Base model: mistralai/Mistral-7B-v0.1
- Adapter: LoRA (q_proj, k_proj, v_proj, o_proj, gate_proj, down_proj, up_proj)
- LoRA Rank: 16, Alpha: 32, Dropout: 0.05
- Batch size: 2 (micro) × 4 (gradient accumulation) = 8 total
- Learning rate: 2e-4
- Optimizer: AdamW
- Scheduler: Cosine
- Sequence length: 2048
- Epochs: 3
- Train on inputs: False (only outputs used for loss)
Hardware
- Trained on RunPod using an H100 PCIe GPU
Framework Versions
- PEFT: 0.14.0
- Transformers: 4.49.0
- PyTorch: 2.5.1+cu124
- Datasets: 3.2.0
- Tokenizers: 0.21.0
- Axolotl: 0.8.0.dev0
License
This model is provided under a custom license:
✅ Use allowed for testing and evaluation
❌ No fine-tuning
❌ No redistribution
❌ No use in commercial or production applications
Contact
For questions or portfolio purposes, contact
project@ezrahernowo.com.
Model Card Authors
Ezra H