tartuNLP / llama-estllm-prototype-0825

EstLLM Prototype 0825

llama-estllm-protype-0825 is the first artifact produced by the EstLLM project. The intention of this release is to evaluate the first prototype in a conversational ChatbotArena-style setting on baromeeter.ai , and thus establish a baseline for future improvements.

The model underwent continuous pre-training starting from Llama-3.1-8B on approximately 35B tokens, then supervised fine-tuning and direct preference optimization were applied.

Use with transformers

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_name = "tartuNLP/llama-estllm-prototype-0825"

model = AutoModelForCausalLM.from_pretrained(
    model_name,
    dtype="auto",
    device_map="auto"
)

# to use on apple silicon, load the following way
# model = AutoModelForCausalLM.from_pretrained(
#     model_name,
#     dtype=torch.float16,
#     device_map="mps",
# )

tokenizer = AutoTokenizer.from_pretrained(model_name)

messages = [
    {"role": "user", "content": "Kas sa räägid eesti keelt?"}
]

text = tokenizer.apply_chat_template(
    messages,
    tokenize=False,
    add_generation_prompt=True
)

model_inputs = tokenizer(text, return_tensors="pt").to(model.device)

generated_ids = model.generate(
    **model_inputs,
    max_new_tokens=128,
    do_sample=True,
    temperature=0.4,
    # specify eos token to stop at the end of the assistant response
    eos_token_id=tokenizer.eos_token_id,
)

# generated_ids include the input tokens as well, so we only decode new tokens
response = tokenizer.decode(
    generated_ids[0][model_inputs["input_ids"].shape[1]:],
    skip_special_tokens=True,
)

print(response)

Model Details

Model Description

• Developed by: TartuNLP and TalTechNLP research groups
• Funded by: Estonian Ministry of Education and Research, “Estonian Language Technology Program 2018-2027”
• Model type: Causal Language Model, Instruction-following
• Language(s) (NLP): Estonian, English
• License: Llama 3.1 Community License Agreement
• Finetuned from model meta-llama/Llama-3.1-8B

Evaluation

Logits-based

Scores for logits-based evaluation benchmarks are available on the EuroEval leaderboard.

Generative

Every benchmark in this category is treated as a generative problem, and thus the evaluation is performed on the model responses obtained with 0 temperature (not logits). The top scores are higlighted with bold . Second best scores are highlighted with italic bold . Rows are sorted in descending order based on the number of parameters of models (not scores). The test set is used for evaluation of each dataset unless noted otherwise.

Instruction-following

Instruction level strict accuracy is reported for IFEval-et.

Multiple Choice

All datasets except Winogrande-et are evaluated in 0-shot mode. Winogrande-et is evaluated in 3-shot mode. Exact match accuracy is reported for every dataset.

Translation

English to Estonian

Limitations

This is an early prototype version. Accordingly, it has limitations in addition to the base Llama limitations:

• Relatively short context of 4096 tokens. It's not expected to perform well on context sizes beyond that.
• Multi-turn conversations are not supported in this version.
• Trained with the original Llama 3.1 system prompt that has a hard-coded date cut-off.

Citation

TBA