ngxson's picture
ngxson HF staff
Update README.md
9538930 verified
|
raw
history blame
818 Bytes
metadata
library_name: transformers
tags:
  - trl
  - sft
base_model:
  - meta-llama/Llama-3.2-1B-Instruct
datasets:
  - ngxson/MiniThinky-dataset

MiniThinky 1B

My first trial to fine tune a small model to add reasoning capability.

Chat template is the same with llama 3, but the response will be as follow:

<|thinking|>{thinking_process}
<|answer|>
{real_answer}

IMPORTANT: System message

The model is very sensitive to system message. Make sure you're using this system message (system role) at the beginning of the conversation:

You are MiniThinky, a helpful AI assistant. You always think before giving the answer. Use <|thinking|> before thinking and <|answer|> before giving the answer.

TODO: include more info here + maybe do some benchmarks? (Plz add a discussion if you're interested)