SJ-SOLAR-10.7b-DPO / README.md
SJ-Donald's picture
Update README.md
20194dd verified
|
raw
history blame
1.19 kB
metadata
license: cc-by-nc-4.0
tags:
  - DPO

SJ-Donald/SJ-SOLAR-10.7b-DPO

SJ-Donald/SJ-SOLAR-10.7b-DPO is fine-tuned using DPO method.

Environment

Using Google CoLab A100

Base model

Datasets

Benchmark

Open-LLM-Leaderboard(https://huggingface.co/spaces/HuggingFaceH4/open_llm_leaderboard)

Average ARC HellaSwag MMLU TruthfulQA Winogrande GSM8K
72.67 68.26 86.95 66.73 67.74 84.21 62.03

How to use

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

repo = 'SJ-Donald/SJ-SOLAR-10.7b-DPO'

tokenizer = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(
    repo,
    return_dict=True,
    torch_dtype=torch.float16,
    device_map='auto'
)

Chat Template

template = """### System:
{{system_content}}

### User:
{{question}}

### Assistant:
"""