gpt-sw3

Model type	gpt-sw3
Class	GPT2CausalLMModel
Task	text-generation
Source	models/gpt2.py

Description

GPT-2 causal language model.

Usage

mobius build --model <MODEL_ID> output_dir/

from mobius import build

model = build("<MODEL_ID>")