Views
No views yet
| Metric | GPT-2-dolly | GPT-2 (base) |
|---|---|---|
| Avg. | 30.91 | 29.99 |
| ARC (25-shot) | 22.70 | 21.84 |
| HellaSwag (10-shot) | 30.15 | 31.6 |
| MMLU (5-shot) | 25.81 | 25.86 |
| TruthfulQA (0-shot) | 44.97 | 40.67 |
1# Use a pipeline as a high-level helper
2>>> from transformers import pipeline
3>>> pipe = pipeline("text-generation", model="lgaalves/gpt2-dolly")
4>>> question = "What is a large language model?"
5>>> answer = pipe(question)
6>>> print(answer[0]['generated_text'])1# Load model directly
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4tokenizer = AutoTokenizer.from_pretrained("lgaalves/gpt2-dolly")
5model = AutoModelForCausalLM.from_pretrained("lgaalves/gpt2-dolly")lgaalves/gpt2-dolly trained using the Databricks Dolly dataset databricks/databricks-dolly-15k.lgaalves/gpt2-dolly was instruction fine-tuned using LoRA on 1 T4 GPU on Google Colab. It took about 1.5 hours to train it.| Metric | Value |
|---|---|
| Avg. | 25.53 |
| ARC (25-shot) | 22.7 |
| HellaSwag (10-shot) | 30.15 |
| MMLU (5-shot) | 25.81 |
| TruthfulQA (0-shot) | 44.97 |
| Winogrande (5-shot) | 51.46 |
| GSM8K (5-shot) | 0.15 |
| DROP (3-shot) | 3.45 |