Struct Gpt2Config 

pub struct Gpt2Config {
    pub n_layer: usize,
    pub n_head: usize,
    pub n_embd: usize,
    pub n_ctx: usize,
    pub vocab_size: usize,
    pub layer_norm_epsilon: f32,
    pub eos_token_id: Option<u32>,
}

Fields

n_layer: usize

n_head: usize

n_embd: usize

n_ctx: usize

vocab_size: usize

layer_norm_epsilon: f32

eos_token_id: Option<u32>

Implementations

impl Gpt2Config

pub fn gpt2() -> Self

GPT-2 small (124M).

pub fn from_json(path: impl AsRef<Path>) -> Result<Self>

pub fn from_json_str(text: &str) -> Result<Self>

Parse an HF config.json string — the primary form; on wasm the string arrives from an HTTP fetch.

Trait Implementations

impl Clone for Gpt2Config

fn clone(&self) -> Gpt2Config

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Gpt2Config

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.