|
---
|
|
license: apache-2.0
|
|
pipeline_tag: text-generation
|
|
library_name: transformers
|
|
language: [
|
|
'en', 'am', 'ar', 'as', 'az', 'be', 'bg', 'bn', 'br', 'bs', 'ca', 'cs', 'cy', 'da', 'de', 'el',
|
|
'eo', 'es', 'et', 'eu', 'fa', 'ff', 'fi', 'fr', 'fy', 'ga', 'gd', 'gl', 'gn', 'gu', 'ha', 'he',
|
|
'hi', 'hr', 'ht', 'hu', 'hy', 'id', 'ig', 'is', 'it', 'ja', 'jv', 'ka', 'kk', 'km', 'kn', 'ko',
|
|
'ku', 'ky', 'la', 'lg', 'li', 'ln', 'lo', 'lt', 'lv', 'mg', 'mk', 'ml', 'mn', 'mr', 'ms', 'my',
|
|
'ne', 'nl', 'no', 'ns', 'om', 'or', 'pa', 'pl', 'ps', 'pt', 'qu', 'rm', 'ro', 'ru', 'sa', 'si',
|
|
'sc', 'sd', 'sk', 'sl', 'so', 'sq', 'sr', 'ss', 'su', 'sv', 'sw', 'ta', 'te', 'th', 'tl', 'tn',
|
|
'tr', 'ug', 'uk', 'ur', 'uz', 'vi', 'wo', 'xh', 'yi', 'yo', 'zu',
|
|
]
|
|
datasets: [
|
|
'yahma/alpaca-cleaned',
|
|
'gbharti/wealth-alpaca_lora',
|
|
'databricks/databricks-dolly-15k',
|
|
'VMware/open-instruct',
|
|
'saillab/taco-datasets',
|
|
'xu-song/cc100-samples',
|
|
'jordiclive/wikipedia-summary-dataset',
|
|
'bigcode/the-stack-smol-xs',
|
|
'm-a-p/CodeFeedback-Filtered-Instruction',
|
|
'jtatman/python-code-dataset-500k',
|
|
'iamtarun/python_code_instructions_18k_alpaca',
|
|
'HuggingFaceH4/CodeAlpaca_20K',
|
|
'cognitivecomputations/dolphin-coder',
|
|
'fblgit/simple-math',
|
|
'gair-prox/open-web-math-pro',
|
|
'rvv-karma/Math-QA',
|
|
'ajibawa-2023/Maths-College',
|
|
'microsoft/orca-math-word-problems-200k',
|
|
'meta-math/MetaMathQA',
|
|
'TIGER-Lab/MathInstruct',
|
|
'TIGER-Lab/WebInstructSub',
|
|
'SkunkworksAI/reasoning-0.01',
|
|
'KingNish/reasoning-base-20k',
|
|
'Magpie-Align/Magpie-Reasoning-150K',
|
|
'thesven/gsm8k-reasoning',
|
|
'AlgorithmicResearchGroup/math_reasoning_autoformalization_track',
|
|
'badrex/llm-emoji-dataset',
|
|
]
|
|
tags:
|
|
- litgpt
|
|
- litdata
|
|
---
|
|
|
|
# tangled-llama-t-32k-base-v0.1
|
|
|
|
![logo](./misc/logo.png)
|
|
|
|
A pretrained language model based on the Llama model with about **25M** parameters. This model has been trained on **22.1B** (`22,111,299,936`) tokens from more than **3.6M** (`3,597,088`) dataset rows.
|
|
|
|
This model **isn't** designed for immediate use but rather for Continued Pretraining and Finetuning on a downstream task. While it can handle a context length of up to **128K** (`131,072`) tokens, it was pretrained with sequences of **2K** (`2048`) tokens.
|
|
|
|
The objective is to streamline the cognitive or reasoning core, eliminating any redundant knowledge from the model.
|
|
|
|
[loss, val_loss](https://api.wandb.ai/links/mtasic85/t66yvgeh)
|
|
|
|
[val_ppl](https://api.wandb.ai/links/mtasic85/osr62qqd)
|
|
|
|
[epoch](https://api.wandb.ai/links/mtasic85/pw0ilz5s)
|
|
|
|
[learning_rate](https://api.wandb.ai/links/mtasic85/867ueoyx)
|
|
|
|
## lm-evaluation-harness
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'hellaswag,gsm8k,truthfulqa_mc2,mmlu,winogrande,arc_challenge' --out_dir 'evaluate-quick/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
Tasks |Version| Filter |n-shot| Metric | |Value | |Stderr|
|
|
|---------------------------------------|------:|----------------|-----:|-----------|---|-----:|---|-----:|
|
|
|arc_challenge | 1|none | 0|acc |↑ |0.1971|± |0.0116|
|
|
| | |none | 0|acc_norm |↑ |0.2423|± |0.0125|
|
|
|gsm8k | 3|flexible-extract| 5|exact_match|↑ |0.0099|± |0.0027|
|
|
| | |strict-match | 5|exact_match|↑ |0.0000|± |0.0000|
|
|
|hellaswag | 1|none | 0|acc |↑ |0.2608|± |0.0044|
|
|
| | |none | 0|acc_norm |↑ |0.2665|± |0.0044|
|
|
|mmlu | 2|none | |acc |↑ |0.2451|± |0.0036|
|
|
| - humanities | 2|none | |acc |↑ |0.2470|± |0.0063|
|
|
| - formal_logic | 1|none | 0|acc |↑ |0.3254|± |0.0419|
|
|
| - high_school_european_history | 1|none | 0|acc |↑ |0.2545|± |0.0340|
|
|
| - high_school_us_history | 1|none | 0|acc |↑ |0.2745|± |0.0313|
|
|
| - high_school_world_history | 1|none | 0|acc |↑ |0.2194|± |0.0269|
|
|
| - international_law | 1|none | 0|acc |↑ |0.2231|± |0.0380|
|
|
| - jurisprudence | 1|none | 0|acc |↑ |0.2685|± |0.0428|
|
|
| - logical_fallacies | 1|none | 0|acc |↑ |0.2025|± |0.0316|
|
|
| - moral_disputes | 1|none | 0|acc |↑ |0.2457|± |0.0232|
|
|
| - moral_scenarios | 1|none | 0|acc |↑ |0.2670|± |0.0148|
|
|
| - philosophy | 1|none | 0|acc |↑ |0.1865|± |0.0221|
|
|
| - prehistory | 1|none | 0|acc |↑ |0.2500|± |0.0241|
|
|
| - professional_law | 1|none | 0|acc |↑ |0.2523|± |0.0111|
|
|
| - world_religions | 1|none | 0|acc |↑ |0.1871|± |0.0299|
|
|
| - other | 2|none | |acc |↑ |0.2456|± |0.0077|
|
|
| - business_ethics | 1|none | 0|acc |↑ |0.3400|± |0.0476|
|
|
| - clinical_knowledge | 1|none | 0|acc |↑ |0.2113|± |0.0251|
|
|
| - college_medicine | 1|none | 0|acc |↑ |0.2543|± |0.0332|
|
|
| - global_facts | 1|none | 0|acc |↑ |0.1800|± |0.0386|
|
|
| - human_aging | 1|none | 0|acc |↑ |0.1749|± |0.0255|
|
|
| - management | 1|none | 0|acc |↑ |0.3398|± |0.0469|
|
|
| - marketing | 1|none | 0|acc |↑ |0.2479|± |0.0283|
|
|
| - medical_genetics | 1|none | 0|acc |↑ |0.3100|± |0.0465|
|
|
| - miscellaneous | 1|none | 0|acc |↑ |0.2171|± |0.0147|
|
|
| - nutrition | 1|none | 0|acc |↑ |0.2647|± |0.0253|
|
|
| - professional_accounting | 1|none | 0|acc |↑ |0.2270|± |0.0250|
|
|
| - professional_medicine | 1|none | 0|acc |↑ |0.2978|± |0.0278|
|
|
| - virology | 1|none | 0|acc |↑ |0.3133|± |0.0361|
|
|
| - social sciences | 2|none | |acc |↑ |0.2584|± |0.0079|
|
|
| - econometrics | 1|none | 0|acc |↑ |0.2193|± |0.0389|
|
|
| - high_school_geography | 1|none | 0|acc |↑ |0.2677|± |0.0315|
|
|
| - high_school_government_and_politics| 1|none | 0|acc |↑ |0.2435|± |0.0310|
|
|
| - high_school_macroeconomics | 1|none | 0|acc |↑ |0.2538|± |0.0221|
|
|
| - high_school_microeconomics | 1|none | 0|acc |↑ |0.2647|± |0.0287|
|
|
| - high_school_psychology | 1|none | 0|acc |↑ |0.2679|± |0.0190|
|
|
| - human_sexuality | 1|none | 0|acc |↑ |0.3435|± |0.0416|
|
|
| - professional_psychology | 1|none | 0|acc |↑ |0.2190|± |0.0167|
|
|
| - public_relations | 1|none | 0|acc |↑ |0.2091|± |0.0390|
|
|
| - security_studies | 1|none | 0|acc |↑ |0.2980|± |0.0293|
|
|
| - sociology | 1|none | 0|acc |↑ |0.2836|± |0.0319|
|
|
| - us_foreign_policy | 1|none | 0|acc |↑ |0.3000|± |0.0461|
|
|
| - stem | 2|none | |acc |↑ |0.2287|± |0.0075|
|
|
| - abstract_algebra | 1|none | 0|acc |↑ |0.2100|± |0.0409|
|
|
| - anatomy | 1|none | 0|acc |↑ |0.2000|± |0.0346|
|
|
| - astronomy | 1|none | 0|acc |↑ |0.2434|± |0.0349|
|
|
| - college_biology | 1|none | 0|acc |↑ |0.3333|± |0.0394|
|
|
| - college_chemistry | 1|none | 0|acc |↑ |0.3000|± |0.0461|
|
|
| - college_computer_science | 1|none | 0|acc |↑ |0.2600|± |0.0441|
|
|
| - college_mathematics | 1|none | 0|acc |↑ |0.3100|± |0.0465|
|
|
| - college_physics | 1|none | 0|acc |↑ |0.2353|± |0.0422|
|
|
| - computer_security | 1|none | 0|acc |↑ |0.2300|± |0.0423|
|
|
| - conceptual_physics | 1|none | 0|acc |↑ |0.2085|± |0.0266|
|
|
| - electrical_engineering | 1|none | 0|acc |↑ |0.2621|± |0.0366|
|
|
| - elementary_mathematics | 1|none | 0|acc |↑ |0.2011|± |0.0206|
|
|
| - high_school_biology | 1|none | 0|acc |↑ |0.2097|± |0.0232|
|
|
| - high_school_chemistry | 1|none | 0|acc |↑ |0.2217|± |0.0292|
|
|
| - high_school_computer_science | 1|none | 0|acc |↑ |0.2300|± |0.0423|
|
|
| - high_school_mathematics | 1|none | 0|acc |↑ |0.1926|± |0.0240|
|
|
| - high_school_physics | 1|none | 0|acc |↑ |0.2318|± |0.0345|
|
|
| - high_school_statistics | 1|none | 0|acc |↑ |0.1806|± |0.0262|
|
|
| - machine_learning | 1|none | 0|acc |↑ |0.2857|± |0.0429|
|
|
|truthfulqa_mc2 | 2|none | 0|acc |↑ |0.4880|± |0.0161|
|
|
|winogrande | 1|none | 0|acc |↑ |0.5185|± |0.0140|
|
|
|
|
| Groups |Version|Filter|n-shot|Metric| |Value | |Stderr|
|
|
|------------------|------:|------|------|------|---|-----:|---|-----:|
|
|
|mmlu | 2|none | |acc |↑ |0.2451|± |0.0036|
|
|
| - humanities | 2|none | |acc |↑ |0.2470|± |0.0063|
|
|
| - other | 2|none | |acc |↑ |0.2456|± |0.0077|
|
|
| - social sciences| 2|none | |acc |↑ |0.2584|± |0.0079|
|
|
| - stem | 2|none | |acc |↑ |0.2287|± |0.0075|
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'leaderboard' --out_dir 'evaluate-leaderboard/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
| Tasks |Version|Filter|n-shot| Metric | |Value | |Stderr|
|
|
|-----------------------------------------------------------|-------|------|-----:|-----------------------|---|-----:|---|------|
|
|
|leaderboard | N/A| | | | | | | |
|
|
| - leaderboard_bbh | N/A| | | | | | | |
|
|
| - leaderboard_bbh_boolean_expressions | 1|none | 3|acc_norm |↑ |0.4600|± |0.0316|
|
|
| - leaderboard_bbh_causal_judgement | 1|none | 3|acc_norm |↑ |0.5027|± |0.0367|
|
|
| - leaderboard_bbh_date_understanding | 1|none | 3|acc_norm |↑ |0.1720|± |0.0239|
|
|
| - leaderboard_bbh_disambiguation_qa | 1|none | 3|acc_norm |↑ |0.2960|± |0.0289|
|
|
| - leaderboard_bbh_formal_fallacies | 1|none | 3|acc_norm |↑ |0.4880|± |0.0317|
|
|
| - leaderboard_bbh_geometric_shapes | 1|none | 3|acc_norm |↑ |0.0000|± | 0|
|
|
| - leaderboard_bbh_hyperbaton | 1|none | 3|acc_norm |↑ |0.5160|± |0.0317|
|
|
| - leaderboard_bbh_logical_deduction_five_objects | 1|none | 3|acc_norm |↑ |0.2000|± |0.0253|
|
|
| - leaderboard_bbh_logical_deduction_seven_objects | 1|none | 3|acc_norm |↑ |0.1480|± |0.0225|
|
|
| - leaderboard_bbh_logical_deduction_three_objects | 1|none | 3|acc_norm |↑ |0.3160|± |0.0295|
|
|
| - leaderboard_bbh_movie_recommendation | 1|none | 3|acc_norm |↑ |0.2360|± |0.0269|
|
|
| - leaderboard_bbh_navigate | 1|none | 3|acc_norm |↑ |0.4680|± |0.0316|
|
|
| - leaderboard_bbh_object_counting | 1|none | 3|acc_norm |↑ |0.0480|± |0.0135|
|
|
| - leaderboard_bbh_penguins_in_a_table | 1|none | 3|acc_norm |↑ |0.1918|± |0.0327|
|
|
| - leaderboard_bbh_reasoning_about_colored_objects | 1|none | 3|acc_norm |↑ |0.1440|± |0.0222|
|
|
| - leaderboard_bbh_ruin_names | 1|none | 3|acc_norm |↑ |0.2360|± |0.0269|
|
|
| - leaderboard_bbh_salient_translation_error_detection | 1|none | 3|acc_norm |↑ |0.1360|± |0.0217|
|
|
| - leaderboard_bbh_snarks | 1|none | 3|acc_norm |↑ |0.5225|± |0.0375|
|
|
| - leaderboard_bbh_sports_understanding | 1|none | 3|acc_norm |↑ |0.4560|± |0.0316|
|
|
| - leaderboard_bbh_temporal_sequences | 1|none | 3|acc_norm |↑ |0.2960|± |0.0289|
|
|
| - leaderboard_bbh_tracking_shuffled_objects_five_objects | 1|none | 3|acc_norm |↑ |0.2120|± |0.0259|
|
|
| - leaderboard_bbh_tracking_shuffled_objects_seven_objects| 1|none | 3|acc_norm |↑ |0.1840|± |0.0246|
|
|
| - leaderboard_bbh_tracking_shuffled_objects_three_objects| 1|none | 3|acc_norm |↑ |0.3160|± |0.0295|
|
|
| - leaderboard_bbh_web_of_lies | 1|none | 3|acc_norm |↑ |0.5200|± |0.0317|
|
|
| - leaderboard_gpqa | N/A| | | | | | | |
|
|
| - leaderboard_gpqa_diamond | 1|none | 0|acc_norm |↑ |0.2172|± |0.0294|
|
|
| - leaderboard_gpqa_extended | 1|none | 0|acc_norm |↑ |0.2454|± |0.0184|
|
|
| - leaderboard_gpqa_main | 1|none | 0|acc_norm |↑ |0.2478|± |0.0204|
|
|
| - leaderboard_ifeval | 3|none | 0|inst_level_loose_acc |↑ |0.1727|± | N/A|
|
|
| | |none | 0|inst_level_strict_acc |↑ |0.1559|± | N/A|
|
|
| | |none | 0|prompt_level_loose_acc |↑ |0.0832|± |0.0119|
|
|
| | |none | 0|prompt_level_strict_acc|↑ |0.0795|± |0.0116|
|
|
| - leaderboard_math_hard | N/A| | | | | | | |
|
|
| - leaderboard_math_algebra_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_math_counting_and_prob_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_math_geometry_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_math_intermediate_algebra_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_math_num_theory_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_math_prealgebra_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_math_precalculus_hard | 1|none | 4|exact_match |↑ |0.0000|± | 0|
|
|
| - leaderboard_mmlu_pro | 0.1|none | 5|acc |↑ |0.1135|± |0.0029|
|
|
| - leaderboard_musr | N/A| | | | | | | |
|
|
| - leaderboard_musr_murder_mysteries | 1|none | 0|acc_norm |↑ |0.5240|± |0.0316|
|
|
| - leaderboard_musr_object_placements | 1|none | 0|acc_norm |↑ |0.2734|± |0.0279|
|
|
| - leaderboard_musr_team_allocation | 1|none | 0|acc_norm |↑ |0.3000|± |0.0290|
|
|
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'bbh_zeroshot,bbh_fewshot,bbh_cot_fewshot,bbh_cot_zeroshot' --out_dir 'evaluate-bigbenchhard/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'mmlu,mmlu_pro' --out_dir 'evaluate-mmlu/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'arc_challenge,boolq,gpqa,hellaswag,openbookqa,piqa,truthfulqa_mc2,winogrande' --out_dir 'evaluate-reasoning/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'mmlu_multilingual,mgsm' --out_dir 'evaluate-multilinguals/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'gsm8k,mathqa' --out_dir 'evaluate-math/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|
|
|Tasks |Version| Filter |n-shot| Metric | |Value | |Stderr|
|
|
|------|------:|----------------|-----:|-----------|---|-----:|---|-----:|
|
|
|gsm8k | 3|flexible-extract| 5|exact_match|↑ |0.0099|± |0.0027|
|
|
| | |strict-match | 5|exact_match|↑ |0.0000|± |0.0000|
|
|
|mathqa| 1|none | 0|acc |↑ |0.2121|± |0.0075|
|
|
| | |none | 0|acc_norm |↑ |0.2114|± |0.0075|
|
|
|
|
|
|
```bash
|
|
litgpt evaluate --tasks 'wikitext,qasper' --out_dir 'evaluate-long/' --batch_size 4 --dtype 'bfloat16' out/pretrain/final/
|
|
```
|
|
|