|
--- |
|
license: llama2 |
|
datasets: |
|
- garage-bAInd/Open-Platypus |
|
pipeline_tag: text-generation |
|
language: |
|
- en |
|
--- |
|
|
|
|
|
# llama-2-13b-hf-platypus |
|
|
|
|
|
**llama-2-13b-hf-platypus** is an instruction fine-tuned model based on the LLaMA2-7B transformer architecture. |
|
|
|
|
|
### Benchmark Metrics |
|
|
|
|
|
| Metric | llama-2-13b-hf-platypus | garage-bAInd/Platypus2-13B| llama-2-13b-hf (base) | |
|
|-----------------------|-------|-------|-------| |
|
| Avg. | 59.71 |**61.35**| 58.66 | |
|
| ARC (25-shot) | 58.87|**61.26**| 59.39 | |
|
| HellaSwag (10-shot) | 82.14|**82.56**| 82.13 | |
|
| MMLU (5-shot) | 54.98|**56.7**| 55.77 | |
|
| TruthfulQA (0-shot) | 42.84|**44.86**| 37.38 | |
|
|
|
|
|
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:** **llama-2-13b-hf-platypus** is an auto-regressive language model based on the LLaMA2 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/llama-2-13b-hf-platypus") |
|
>>> question = "What is a large language model?" |
|
>>> answer = pipe(question) |
|
>>> print(answer[0]['generated_text']) |
|
|
|
``` |
|
|
|
or, you can load the model direclty using: |
|
|
|
```python |
|
# Load model directly |
|
from transformers import AutoTokenizer, AutoModelForCausalLM |
|
|
|
tokenizer = AutoTokenizer.from_pretrained("lgaalves/llama-2-13b-hf-platypus") |
|
model = AutoModelForCausalLM.from_pretrained("lgaalves/llama-2-13b-hf-platypus") |
|
``` |
|
|
|
### Training Dataset |
|
|
|
`lgaalves/llama-2-13b-hf-platypus` trained using STEM and logic based dataset [`garage-bAInd/Open-Platypus`](https://huggingface.co/datasets/garage-bAInd/Open-Platypus). |
|
|
|
|
|
### Training Procedure |
|
|
|
`lgaalves/llama-2-13b-hf-platypus` was instruction fine-tuned using LoRA on 1 A100-40GB and took about 2.5 hours for completing the training. |
|
|
|
|
|
### Limitations and bias |
|
|
|
Llama 2 and fine-tuned variants are a new technology that carries risks with use. Testing conducted to date has been in English, and has not covered, nor could it cover all scenarios. For these reasons, as with all LLMs, Llama 2 and any fine-tuned varient's potential outputs cannot be predicted in advance, and the model may in some instances produce inaccurate, biased or other objectionable responses to user prompts. Therefore, before deploying any applications of Llama 2 variants, developers should perform safety testing and tuning tailored to their specific applications of the model. |
|
|
|
Please see the Responsible Use Guide available at https://ai.meta.com/llama/responsible-use-guide/ |
