lgaalves's picture
Adding Evaluation Results (#1)
f58d636
|
raw
history blame
3.52 kB
metadata
license: apache-2.0
datasets:
  - garage-bAInd/Open-Platypus
pipeline_tag: text-generation
language:
  - en

mistral-7b_open_platypus

mistral-7b_open_platypus is an instruction fine-tuned model based on the Mistral-7B transformer architecture.

Benchmark Metrics

Metric mistral-7b_open_platypus mistralai/Mistral-7B-v0.1 garage-bAInd/Platypus2-7B
Avg. - 62.40 56.13
ARC (25-shot) - 59.98 55.20
HellaSwag (10-shot) - 83.31 78.84
MMLU (5-shot) - 64.16 49.83
TruthfulQA (0-shot) - 42.15 40.64

We use state-of-the-art Language Model 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: mistral-7b_open_platypus is an auto-regressive language model based on the Mistral-7B transformer architecture.
  • Language(s): English

How to use:

# Use a pipeline as a high-level helper
>>> from transformers import pipeline
>>> pipe = pipeline("text-generation", model="lgaalves/mistral-7b_open_platypus")
>>> question = "What is a large language model?"
>>> answer = pipe(question)
>>> print(answer[0]['generated_text'])

or, you can load the model direclty using:

# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("lgaalves/mistral-7b_open_platypus")
model = AutoModelForCausalLM.from_pretrained("lgaalves/mistral-7b_open_platypus")

Prompt format

 "<s>[INST] What is your favourite condiment? [/INST]"

Training Dataset

lgaalves/mistral-7b_open_platypus trained using STEM and logic based dataset garage-bAInd/Open-Platypus.

Training Procedure

lgaalves/mistral-7b_open_platypus was instruction fine-tuned using LoRA on 1 Tesla V100-SXM2-16GB. In total, it took 11 hours to fine tune the model.

Limitations and bias

Mistral 7B 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.

Open LLM Leaderboard Evaluation Results

Detailed results can be found here

Metric Value
Avg. 49.19
ARC (25-shot) 55.8
HellaSwag (10-shot) 82.13
MMLU (5-shot) 59.76
TruthfulQA (0-shot) 48.87
Winogrande (5-shot) 78.61
GSM8K (5-shot) 12.59
DROP (3-shot) 6.59