Views
No views yet
| Metric | falcon-rw-1b-instruct-openorca | falcon-rw-1b |
|---|---|---|
| ARC | 34.56 | 35.07 |
| HellaSwag | 60.93 | 63.56 |
| MMLU | 28.77 | 25.28 |
| TruthfulQA | 37.42 | 35.96 |
| Winogrande | 60.69 | 62.04 |
| GSM8K | 3.41 | 0.53 |
| Average | 37.63 | 37.07 |
<SYS>, <INST>, and <RESP> tags to demarcate different parts of the input. The system message and instruction are placed within these tags, with the <RESP> tag triggering the model's response.1from transformers import AutoTokenizer, AutoModelForCausalLM
2import transformers
3import torch
4
5model = 'ericzzz/falcon-rw-1b-instruct-openorca'
6
7tokenizer = AutoTokenizer.from_pretrained(model)
8pipeline = transformers.pipeline(
9 'text-generation',
10 model=model,
11 tokenizer=tokenizer,
12 torch_dtype=torch.bfloat16,
13 device_map='auto',
14)
15
16system_message = 'You are a helpful assistant. Give short answers.'
17instruction = 'What is AI? Give some examples.'
18prompt = f'<SYS> {system_message} <INST> {instruction} <RESP> '
19
20response = pipeline(
21 prompt,
22 max_length=200,
23 repetition_penalty=1.05
24)
25
26print(response[0]['generated_text'])
27# AI, or Artificial Intelligence, refers to the ability of machines and software to perform tasks that require human intelligence, such as learning, reasoning, and problem-solving. It can be used in various fields like computer science, engineering, medicine, and more. Some common applications include image recognition, speech translation, and natural language processing.| Metric | Value |
|---|---|
| Avg. | 37.63 |
| AI2 Reasoning Challenge (25-Shot) | 34.56 |
| HellaSwag (10-Shot) | 60.93 |
| MMLU (5-Shot) | 28.77 |
| TruthfulQA (0-shot) | 37.42 |
| Winogrande (5-shot) | 60.69 |
| GSM8k (5-shot) | 3.41 |