gpt_neox

Model type	gpt_neox
Class	GPTNeoXCausalLMModel
Task	text-generation
Source	models/gpt_neox.py

Description

GPT-NeoX / Pythia causal language model.

Usage

mobius build --model <MODEL_ID> output_dir/

from mobius import build

model = build("<MODEL_ID>")