language:
- en
license: mit
library_name: transformers
tags:
- dpo
- 2b
- gemma
datasets:
- argilla/distilabel-capybara-dpo-7k-binarized
pipeline_tag: text-generation
widget:
- text: |-
3713841893836/4?
Limit your response to mathematical expressions and symbols.
example_title: 'Return only numbers. '
- text: >-
A group of 10 people is split into 3 different committees of 3, 4, and 3
people, respectively. In how many ways can this be done?
example_title: Solve Problem
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.