You can see the process with instructions for creating the model in the notebook: Aligning_DPO_gemma-2b.ipynb
To create it, we started with the google/gemma-2b-it model and applied DPO alignment using the distilabel-capybara-dpo-7k-binarized dataset.
Gemma is a family of lightweight, state-of-the-art open models from Google, built from the same research and technology used to create the Gemini models.
Row in the Dataset:
{'prompt': '<|user|>\nAssist me in calculating 9319357631 plus 595. Numbers and symbols only, please.<|end|>\n<|assistant|>\n', 'chosen': 'The sum of 9319357631 and 595 is 9319358226.<|end|>\n', 'rejected': 'The result of adding 9319357631 and 595 is 9319363626.<|end|>\n'}
Prompt: 3713841893836/4? Limit your response to mathematical expressions and symbols.
Response from the Base model:
To find the result of the division, we can simply divide the given number by 4: $$ \frac{3713841893836}{4} = 928460473459
Response from the fine-tuned model:
3713841893836 ÷ 4 = 928460473459
If you want to see how the model was created, you can check out the repository where the book's notebooks are kept up-to-date.