Edit model card

The license is cc-by-nc-4.0.

GAI-LLM/OPEN-SOLAR-KO-10.7B-dpo-v1

Model Details

Model Developers Donghoon Oh, Hanmin Myung (SK C&C G.AI Eng)

Input Models input text only.

Output Models generate text only.

Model Architecture
GAI-LLM/OPEN-SOLAR-KO-10.7B-dpo-v1 is an auto-regressive language model based on the LLaMA2 transformer architecture.

Base Model GAI-LLM/OPEN-SOLAR-KO-10.7B-mixed-v15

Training Dataset

  • We combined Open Korean Dateset using mixed-strategy with DPO
  • We use A100 GPU 80GB * 8, when training.

Model Benchmark

KO-LLM leaderboard

Implementation Code

### GAI-LLM/OPEN-SOLAR-KO-10.7B-dpo-v1
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

repo = "GAI-LLM/OPEN-SOLAR-KO-10.7B-dpo-v1"
model = AutoModelForCausalLM.from_pretrained(
        repo,
        return_dict=True,
        torch_dtype=torch.float16,
        device_map='auto'
)
tokenizer = AutoTokenizer.from_pretrained(repo)
Downloads last month
1,232