Views
No views yet
pip install sentencepiece1ffrom transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = 'mediocredev/open-llama-3b-v2-chat'
4tokenizer_id = 'mediocredev/open-llama-3b-v2-chat'
5tokenizer = AutoTokenizer.from_pretrained(tokenizer_id)
6model = AutoModelForCausalLM.from_pretrained(model_id)
7
8chat_history = [
9 {"role": "user", "content": "Hello!"},
10 {"role": "assistant", "content": "I am here."},
11 {"role": "user", "content": "How many days are there in a leap year?"},
12]
13
14input_ids = tokenizer.apply_chat_template(
15 chat_history, tokenize=True, add_generation_prompt=True, return_tensors="pt"
16).to(model.device)
17output_tokens = model.generate(
18 input_ids,
19 repetition_penalty=1.05,
20 max_new_tokens=1000,
21)
22output_text = tokenizer.decode(
23 output_tokens[0][len(input_ids[0]) :], skip_special_tokens=True
24)
25
26print(output_text)
27# Assistant: There are 366 days in a leap year, which is one more day than the standard year.| Metric | Value |
|---|---|
| Avg. | 40.93 |
| AI2 Reasoning Challenge (25-Shot) | 40.61 |
| HellaSwag (10-Shot) | 70.30 |
| MMLU (5-Shot) | 28.73 |
| TruthfulQA (0-shot) | 37.84 |
| Winogrande (5-shot) | 65.51 |
| GSM8k (5-shot) | 2.58 |