Downloads · 30 days
4
6% of all-time downloads
Kazchoko/llama3.2-3b-kaz-dpo-lora
llama3.2-3b-kaz-dpo-lora is a machine learning model from Kazchoko. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for peft.
This model tunes Meta Llama 3.2 3B-Instruct using the Direct Preference Optimization (DPO) objective with a LoRA adapter configuration. The goal is to align the base model’s responses with human preference data aligne…
Downloads · 30 days
4
6% of all-time downloads
All-time downloads
69
Public
Repo size
26.4 MB
Likes
0
Public
Click a slice to open those files.
.json17.3 MB · 65%
From the Hugging Face model README
This model tunes Meta Llama 3.2 3B-Instruct using the Direct Preference Optimization (DPO) objective with a LoRA adapter configuration.
The goal is to align the base model’s responses with human preference data aligned to my personal persona and values.
meta-llama/Llama-3.2-3B-Instructtransformers, trl, peft, bitsandbytes, torchOnly intended use is for the HW1 in CS329X
The model was fine-tuned on a preference dataset (PREFERENCE_DATA) selected by Kazunori Fukuhara
Fine-tuning was performed using TRL’s DPOTrainer and PEFT’s LoRA adapters with 4-bit quantization.
| Parameter | Value |
|---|---|
| Base Model | meta-llama/Llama-3.2-3B-Instruct |
| Learning Rate | 1 × 10⁻⁵ |
| Batch Size | 2 |
| Epochs | 3 |
| LoRA r | 8 |
| LoRA α | 16 |
| LoRA Dropout | 0.1 |
| Precision | bfloat16 |
| Quantization | 4-bit NF4 |
| Optimizer | default (AdamW) |
| Trainer | TRL DPOTrainer |
| Hardware | Environment |
|---|---|
| A100 / T4 GPU (16–40 GB) | Google Colab Pro |
| Framework Versions | transformers 4.44+, trl 0.8+, peft 0.10.0, bitsandbytes 0.43+, torch 2.3+ |