Views
No views yet
1from transformers import pipeline
2
3model_tag = "postbot/emailgen-pythia-410m-deduped"
4generator = pipeline(
5 "text-generation",
6 model=model_tag,
7)
8
9prompt = """
10Hello,
11
12Following up on the bubblegum shipment."""
13
14result = generator(
15 prompt,
16) # generate
17print(result[0]["generated_text"])| Metric | Value |
|---|---|
| Avg. | 26.65 |
| ARC (25-shot) | 27.9 |
| HellaSwag (10-shot) | 40.04 |
| MMLU (5-shot) | 27.35 |
| TruthfulQA (0-shot) | 38.2 |
| Winogrande (5-shot) | 52.09 |
| GSM8K (5-shot) | 0.0 |
| DROP (3-shot) | 0.99 |