gpt-neox-20b

Try Try

gpt-neox-20b has 20.7B parameters (20.7B active per token) and takes 41.3 GB on disk. See where its parameters live, layer by layer, and what hardware it needs.

At a glance

EleutherAI/gpt-neox-20b
Repo createdApr 7, 2022updated Jan 31, 2024
Model typeDenseFull attention (MHA)
InputsText
Total parameters20.7B
Active per token20.7B100% of the model
ExpertsNone (dense)
Max context (from config)2K tokens
Layers44
On disk41.3 GB46 files
PrecisionFP16 (100%), other (0.45%)
QuantizationNoneoriginal precision (FP16)
Fewest GPUs1× RTX 6000 Adaa single 48 GB card · weights only
Licenseapache-2.0
GitHubEleutherAI/gpt-neox