Text Generation
Transformers
Safetensors
English
mega
Inference Endpoints
Edit model card

BEE-spoke-data/mega-ar-126m-4k

This may not be the best language model, but it is a language model! It's interesting for several reasons, not the least of which is that it's not technically a transformer.

Details:

  • 768 hidden size, 12 layers
  • no MEGA chunking, 4096 context length
  • EMA dimension 16, shared dimension 192
  • tokenizer: GPT NeoX
  • train-from-scratch

For more info on MEGA (& what some of the params above mean), check out the model docs or the original paper

Usage

Usage is the same as any other small textgen model.

Given the model's small size and architecture, it's probably best to leverage its longer context by adding input context to "see more" rather than "generate more".

evals

Initial data:

hf-causal-experimental (pretrained=BEE-spoke-data/mega-ar-126m-4k,revision=main,trust_remote_code=True,dtype='float'), limit: None, provide_description: False, num_fewshot: 0, batch_size: 4

Task Version Metric Value Stderr
arc_easy 0 acc 0.4415 ± 0.0102
acc_norm 0.3969 ± 0.0100
boolq 1 acc 0.5749 ± 0.0086
lambada_openai 0 ppl 94.9912 ± 3.9682
acc 0.2408 ± 0.0060
openbookqa 0 acc 0.1660 ± 0.0167
acc_norm 0.2780 ± 0.0201
piqa 0 acc 0.5974 ± 0.0114
acc_norm 0.5914 ± 0.0115
winogrande 0 acc 0.4830 ± 0.0140

Downloads last month
3,444
Safetensors
Model size
126M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Datasets used to train BEE-spoke-data/mega-ar-126m-4k