plm

Model type

plm

Class

PLMCausalLMModel

Task

text-generation

Source

models/plm.py

Description

PLM causal LM with one embedding-owned, tied output allocation.

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")