File size: 2,539 Bytes
8b1d820 e0e76b6 8b1d820 190820c 2810604 190820c 2810604 190820c e0e76b6 2810604 472a0a5 2810604 75610e7 2810604 75610e7 2810604 75610e7 2810604 eb173de 2810604 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
---
license: mit
datasets:
- databricks/databricks-dolly-15k
language:
- en
pipeline_tag: text-generation
---
# GPT-2-dolly
**GPT-2-dolly** is an instruction fine-tuned model based on the GPT-2 transformer architecture.
### Benchmark Metrics
| Metric | GPT-2-dolly | GPT-2 (base) |
|-----------------------|-------|-------|
| Avg. | 29.85 | **29.99** |
| ARC (25-shot) | 21.76 | **21.84** |
| HellaSwag (10-shot) | 30.77 | **31.6** |
| MMLU (5-shot) | 24.66 | **25.86** |
| TruthfulQA (0-shot) | **42.22** | 40.67 |
We use state-of-the-art [Language Model Evaluation Harness](https://github.com/EleutherAI/lm-evaluation-harness) to run the benchmark tests above, using the same version as the HuggingFace LLM Leaderboard. Please see below for detailed instructions on reproducing benchmark results.
### Model Details
* **Trained by**: Luiz G A Alves
* **Model type:** **GPT-2-dolly** is an auto-regressive language model based on the GPT-2 transformer architecture.
* **Language(s)**: English
### How to use:
```python
# Use a pipeline as a high-level helper
>>> from transformers import pipeline
>>> pipe = pipeline("text-generation", model="lgaalves/gpt2-dolly")
>>> question = "What is a large language model?"
>>> answer = pipe(question)
>>> print(answer[0]['generated_text'])
What is a large language model?
A large language model aims for understanding a large group of phenomena through computational methods which allow more precise models.
A model also encourages the use of empirical concepts such as equations, models, natural numbers, natural language
```
or, you can load the model direclty using:
```python
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("lgaalves/gpt2-dolly")
model = AutoModelForCausalLM.from_pretrained("lgaalves/gpt2-dolly")
```
### Training Dataset
`lgaalves/gpt2-dolly` trained using the Databricks Dolly dataset [`databricks/databricks-dolly-15k`](https://huggingface.co./datasets/databricks/databricks-dolly-15k).
### Training Procedure
`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.
# Intended uses, limitations & biases
You can use the raw model for text generation or fine-tune it to a downstream task. The model was not extensively tested and may produce false information. It contains a lot of unfiltered content from the internet, which is far from neutral.
|