Views
No views yet
lm-evaluation-harness. It shows a strong performance in factual knowledge compared to other models in its weight class.| Benchmark | Metric | CoALa-1 (183M) | GPT-2 (124M) | OPT-125M |
|---|---|---|---|---|
| ARC-Easy | acc_norm | 28.87% | 27.00% | 24.50% |
| HellaSwag | acc_norm | 26.96% | 28.50% | 26.00% |

Figure 1: Comparison of ARC-Easy (Knowledge) and HellaSwag (Reasoning) scores. CoALa-1 leads in factual knowledge retrieval among sub-200M parameter models.
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "CocoEntertainment/CoALa-1-Pretuned"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)