Index

A

apply_linear_attention_mask() (in module tptt.modeling_tptt) architectures (tptt.configuration_tptt.TpttConfig attribute) (tptt.TpttConfig attribute)

auto_map (tptt.configuration_tptt.TpttConfig attribute) (tptt.TpttConfig attribute)

C

CausalAvgPool1d (class in tptt.modeling_tptt) chunk_delta_product_forward() (tptt.LinearAttentionOp static method) (tptt.modeling_tptt.LinearAttentionOp static method) chunk_sequence() (in module tptt.modeling_tptt)

compute_gate() (tptt.LinearAttentionOp method) (tptt.modeling_tptt.LinearAttentionOp method) config_class (tptt.modeling_tptt.TpttModel attribute) (tptt.TpttModel attribute) convert_sets_to_lists() (in module tptt.configuration_tptt)

D

describe() (in module tptt.modeling_tptt)

E

ensure_int() (in module tptt.train_tptt) ensure_stability() (in module tptt.modeling_tptt)

expand_virtual_tokens() (in module tptt.modeling_tptt) extract_layer_idx() (in module tptt.modeling_tptt) extract_template_variables() (in module tptt.configuration_tptt)

F

fast_invert_matrix() (in module tptt.modeling_tptt) find_embedding_lm() (in module tptt.modeling_tptt) forward() (tptt.LinearAttention method) (tptt.LinearAttentionOp method) (tptt.LiZAttention method) (tptt.modeling_tptt.CausalAvgPool1d method) (tptt.modeling_tptt.LinearAttention method) (tptt.modeling_tptt.LinearAttentionOp method) (tptt.modeling_tptt.LiZAttention method) (tptt.modeling_tptt.TpttModel method) (tptt.TpttModel method)

from_pretrained() (tptt.modeling_tptt.TpttModel class method) (tptt.TpttModel class method)

G

generate() (tptt.modeling_tptt.TpttModel method) (tptt.TpttModel method) generate_model_card() (in module tptt) (in module tptt.configuration_tptt) get_cache() (tptt.LinearAttentionOp method) (tptt.modeling_tptt.LinearAttentionOp method)

get_mode_name() (in module tptt.configuration_tptt) get_tptt_model() (in module tptt) (in module tptt.modeling_tptt) get_valid_chunk_size() (in module tptt.modeling_tptt)

I

inject_liza_attention() (tptt.modeling_tptt.TpttModel static method) (tptt.TpttModel static method)

L

LCache (class in tptt) (class in tptt.modeling_tptt) LinearAttention (class in tptt) (class in tptt.modeling_tptt) LinearAttentionOp (class in tptt) (class in tptt.modeling_tptt)

LiZACallback (class in tptt) (class in tptt.train_tptt) LiZAttention (class in tptt) (class in tptt.modeling_tptt) load_tptt_safetensors() (in module tptt) (in module tptt.modeling_tptt)

M

match_dim() (in module tptt.modeling_tptt) model_type (tptt.configuration_tptt.TpttConfig attribute) (tptt.TpttConfig attribute) module tptt tptt.configuration_tptt tptt.modeling_tptt tptt.train_tptt

O

on_evaluate() (tptt.SaveBestModelCallback method) (tptt.train_tptt.SaveBestModelCallback method) on_log() (tptt.LiZACallback method) (tptt.train_tptt.LiZACallback method)

on_step_end() (tptt.LiZACallback method) (tptt.train_tptt.LiZACallback method)

P

parse_mode_name() (in module tptt) (in module tptt.configuration_tptt)

R

RECURRENT_MODES (tptt.configuration_tptt.TpttConfig attribute) (tptt.TpttConfig attribute) reset() (tptt.LCache method) (tptt.modeling_tptt.LCache method)

retie_lm_after_load() (tptt.modeling_tptt.TpttModel method) (tptt.TpttModel method)

S

save_cache() (tptt.LinearAttentionOp method) (tptt.modeling_tptt.LinearAttentionOp method) save_pretrained() (tptt.modeling_tptt.TpttModel method) (tptt.TpttModel method)

SaveBestModelCallback (class in tptt) (class in tptt.train_tptt) sequential_delta_product_scan() (in module tptt.modeling_tptt) soft_clamp() (in module tptt.modeling_tptt) split_qkv() (in module tptt.modeling_tptt)

T

tptt module tptt.configuration_tptt module tptt.modeling_tptt module

tptt.train_tptt module TpttConfig (class in tptt) (class in tptt.configuration_tptt) TpttModel (class in tptt) (class in tptt.modeling_tptt) truncate_attention_mask() (in module tptt.modeling_tptt)

U

unlinear_activation() (in module tptt.modeling_tptt)

update() (tptt.LCache method) (tptt.modeling_tptt.LCache method)