Views
No views yet
KETI Llama 7B v0.1 is a long-context causal language model released under the
Hugging Face repository KETI-AIR/keti-llama-7b-v0.1.outputs/llama-8b-keti-dpo-rl-mergedLlamaForCausalLMbfloat1620260604_202553| Category | Dataset | Version | Metric | Mode | Score |
|---|---|---|---|---|---|
| Core | core_average | - | naive_average | gen | 27.77 |
| Instruction Following | IFEval | 353ae7 | Prompt-level-strict-accuracy | gen | 50.65 |
| Math Calculation | aime2024 | bc6078 | accuracy | gen | 16.67 |
| Math Calculation | aime2025 | 5e9f4f | accuracy | gen | 3.33 |
| Math Calculation | math_prm800k_500 | 11c4b5 | accuracy | gen | 60.20 |
| General Reasoning | bbh | - | naive_average | gen | 11.87 |
| General Reasoning | GPQA_diamond | 5aeece | accuracy | gen | 20.71 |
| Knowledge | mmlu_pro | - | naive_average | gen | 28.26 |
| Code | openai_humaneval | dcae0e | humaneval_pass@1 | gen | 60.98 |
| Code | lcb_code_generation | b5b6c5 | pass@1 | gen | 6.00 |
| Long Context Reasoning | leval | - | naive_average | gen | 39.37 |
| Long Context Reasoning | longbench | - | naive_average | gen | 20.57 |
| Long Context Reasoning | LongBenchv2 | 75fbba | accuracy | gen | 24.85 |
| Long Context Reasoning | keti_long_ctx_gutenberg | - | naive_average | gen | 17.62 |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "KETI-AIR/keti-llama-7b-v0.1"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.bfloat16,
10 device_map="auto",
11 trust_remote_code=True,
12)
13
14messages = [
15 {"role": "user", "content": "Explain why long-context reasoning is useful."}
16]
17inputs = tokenizer.apply_chat_template(
18 messages,
19 add_generation_prompt=True,
20 return_tensors="pt",
21).to(model.device)
22
23outputs = model.generate(
24 inputs,
25 max_new_tokens=512,
26 do_sample=True,
27 temperature=0.7,
28 top_p=0.9,
29)
30print(tokenizer.decode(outputs[0][inputs.shape[-1]:], skip_special_tokens=True))