MCPcopy Create free account

hub / github.com/SafeAILab/EAGLE / types & classes

Types & classes129 in github.com/SafeAILab/EAGLE

⨍ Functions 596◇ Types & classes 129↳ Endpoints 4
↓ 5 callersClassQwen3RMSNorm
eagle/model/modeling_qwen3_kv.py:116
↓ 4 callersClassLlamaRMSNorm
eagle/traineagle3/cnets.py:355
↓ 4 callersClassLlamaRMSNorm
eagle/model/cnets.py:370
↓ 4 callersClassQwen3Model
eagle/model/modeling_qwen3_kv.py:421
↓ 3 callersClassDataCollatorWithPadding
eagle/train/main_deepspeed.py:169
↓ 3 callersClassLlamaRMSNorm
LlamaRMSNorm is equivalent to T5LayerNorm. Args: hidden_size (int): The size of the hidden states. eps (float, optional
eagle/traineagle3/modeling_llama_kv.py:104
↓ 3 callersClassLlamaRMSNorm
LlamaRMSNorm is equivalent to T5LayerNorm. Args: hidden_size (int): The size of the hidden states. eps (float, optional): A
eagle/model/modeling_llama_kv.py:104
↓ 3 callersClassMixtralRMSNorm
eagle/model/modeling_mixtral_kv.py:189
↓ 3 callersClassModel
eagle/testbug/model/cnets.py:454
↓ 3 callersClassQwen2Model
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`Qwen2DecoderLayer`] Args: config: Qwen2Config
eagle/model/modeling_qwen2_kv.py:808
↓ 3 callersClassQwen2RMSNorm
eagle/model/modeling_qwen2_kv.py:63
↓ 3 callersClassnode
eagle/modeling_eagle.py:562
↓ 3 callersClassnode
eagle/testbug/model/utils_c.py:32
↓ 3 callersClassnode
eagle/model/utils_c.py:35
↓ 2 callersClassCustomDataset
eagle/train/main_deepspeed.py:125
↓ 2 callersClassCustomDataset
eagle/train/main.py:134
↓ 2 callersClassDataCollatorWithPadding
eagle/traineagle3/main.py:174
↓ 2 callersClassDataCollatorWithPadding
eagle/train/main.py:178
↓ 2 callersClassEAGLERMSNorm
eagle/modeling_eagle.py:481
↓ 2 callersClassKVCache
A key-value cache for the model. This class provides a mechanism to maintain a growing cache of keys and values, particularly useful for
eagle/testbug/model/kv_cache.py:4
↓ 2 callersClassKVCache
A key-value cache for the model. This class provides a mechanism to maintain a growing cache of keys and values, particularly useful for
eagle/model/kv_cache.py:4
↓ 2 callersClassLlamaModel
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`LlamaDecoderLayer`] Args: config: LlamaCo
eagle/traineagle3/modeling_llama_kv.py:978
↓ 2 callersClassLlamaModel
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`LlamaDecoderLayer`] Args: config: LlamaConfig
eagle/model/modeling_llama_kv.py:978
↓ 2 callersClassLlamaRMSNorm
eagle/testbug/model/cnets.py:362
↓ 2 callersClassLlamaRMSNorm
eagle/model/cnets1.py:375
↓ 2 callersClassLlamaRotaryEmbedding
eagle/model/cnets.py:110
↓ 2 callersClassLlamaRotaryEmbedding
eagle/model/cnets1.py:110
↓ 2 callersClassMixtralModel
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`MixtralDecoderLayer`] Args: config: MixtralCo
eagle/model/modeling_mixtral_kv.py:712
↓ 2 callersClassModel
eagle/traineagle3/cnets.py:480
↓ 2 callersClassModel
eagle/model/cnets.py:478
↓ 2 callersClassModel
eagle/model/cnets1.py:472
↓ 2 callersClassQwen2RotaryEmbedding
eagle/model/modeling_qwen2_kv.py:84
↓ 1 callersClassAddGaussianNoise
eagle/train/main_deepspeed.py:100
↓ 1 callersClassAddGaussianNoise
eagle/train/main.py:109
↓ 1 callersClassAddUniformNoise
eagle/train/main_deepspeed.py:113
↓ 1 callersClassAddUniformNoise
eagle/train/main.py:122
↓ 1 callersClassEAGLEAttention
Multi-headed attention from 'Attention Is All You Need' paper
eagle/modeling_eagle.py:308
↓ 1 callersClassEAGLEDecoderLayer
eagle/modeling_eagle.py:495
↓ 1 callersClassEAGLEDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla
eagle/modeling_eagle.py:149
↓ 1 callersClassEAGLELinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev
eagle/modeling_eagle.py:130
↓ 1 callersClassEAGLEMLP
eagle/modeling_eagle.py:447
↓ 1 callersClassEAGLEModel
eagle/modeling_eagle.py:695
↓ 1 callersClassEAGLERotaryEmbedding
eagle/modeling_eagle.py:94
↓ 1 callersClassLlamaAttention
Multi-headed attention from 'Attention Is All You Need' paper
eagle/testbug/model/cnets.py:190
↓ 1 callersClassLlamaAttention
Multi-headed attention from 'Attention Is All You Need' paper
eagle/traineagle3/cnets.py:183
↓ 1 callersClassLlamaAttention
LlamaAttention is a multi-headed attention module based on the 'Attention Is All You Need' paper. Args: config (LlamaConfig): Co
eagle/traineagle3/modeling_llama_kv.py:558
↓ 1 callersClassLlamaAttention
Multi-headed attention from 'Attention Is All You Need' paper
eagle/model/cnets.py:191
↓ 1 callersClassLlamaAttention
Multi-headed attention from 'Attention Is All You Need' paper
eagle/model/cnets1.py:191
↓ 1 callersClassLlamaAttention
LlamaAttention is a multi-headed attention module based on the 'Attention Is All You Need' paper. Args: config (LlamaConfig): Config
eagle/model/modeling_llama_kv.py:558
↓ 1 callersClassLlamaDecoderLayer
LlamaDecoderLayer represents a single layer of the Llama decoder. Args: config (LlamaConfig): Configuration for the decoder laye
eagle/traineagle3/modeling_llama_kv.py:776
↓ 1 callersClassLlamaDecoderLayer
eagle/model/cnets1.py:392
↓ 1 callersClassLlamaDecoderLayer
LlamaDecoderLayer represents a single layer of the Llama decoder. Args: config (LlamaConfig): Configuration for the decoder layer.
eagle/model/modeling_llama_kv.py:776
↓ 1 callersClassLlamaDecoderLayeremb
eagle/traineagle3/cnets.py:372
↓ 1 callersClassLlamaDecoderLayeremb
eagle/model/cnets.py:387
↓ 1 callersClassLlamaDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla
eagle/testbug/model/cnets.py:165
↓ 1 callersClassLlamaDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla
eagle/traineagle3/cnets.py:156
↓ 1 callersClassLlamaDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla.
eagle/traineagle3/modeling_llama_kv.py:345
↓ 1 callersClassLlamaDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla
eagle/model/cnets.py:165
↓ 1 callersClassLlamaDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla
eagle/model/cnets1.py:165
↓ 1 callersClassLlamaDynamicNTKScalingRotaryEmbedding
LlamaRotaryEmbedding extended with Dynamic NTK scaling. Credits to the Reddit users /u/bloc97 and /u/emozilla.
eagle/model/modeling_llama_kv.py:345
↓ 1 callersClassLlamaLinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev
eagle/testbug/model/cnets.py:146
↓ 1 callersClassLlamaLinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev
eagle/traineagle3/cnets.py:137
↓ 1 callersClassLlamaLinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. This class adds linear scaling to LlamaRotaryEmbedding. Credits to the Reddit user /u/
eagle/traineagle3/modeling_llama_kv.py:294
↓ 1 callersClassLlamaLinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev
eagle/model/cnets.py:146
↓ 1 callersClassLlamaLinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. Credits to the Reddit user /u/kaiokendev
eagle/model/cnets1.py:146
↓ 1 callersClassLlamaLinearScalingRotaryEmbedding
LlamaRotaryEmbedding extended with linear scaling. This class adds linear scaling to LlamaRotaryEmbedding. Credits to the Reddit user /u/kai
eagle/model/modeling_llama_kv.py:294
↓ 1 callersClassLlamaMLP
eagle/testbug/model/cnets.py:329
↓ 1 callersClassLlamaMLP
eagle/traineagle3/cnets.py:317
↓ 1 callersClassLlamaMLP
LlamaMLP is a multi-layer perceptron module used in the Llama model. Args: config: The configuration for the MLP. Attribu
eagle/traineagle3/modeling_llama_kv.py:473
↓ 1 callersClassLlamaMLP
eagle/model/cnets.py:336
↓ 1 callersClassLlamaMLP
eagle/model/cnets1.py:341
↓ 1 callersClassLlamaMLP
LlamaMLP is a multi-layer perceptron module used in the Llama model. Args: config: The configuration for the MLP. Attributes:
eagle/model/modeling_llama_kv.py:473
↓ 1 callersClassLlamaRotaryEmbedding
eagle/testbug/model/cnets.py:111
↓ 1 callersClassLlamaRotaryEmbedding
eagle/traineagle3/cnets.py:101
↓ 1 callersClassLlamaRotaryEmbedding
Llama Rotary Positional Embedding Module. Args: dim (int): The dimension of the embedding. max_position_embeddings (int
eagle/traineagle3/modeling_llama_kv.py:135
↓ 1 callersClassLlamaRotaryEmbedding
Llama Rotary Positional Embedding Module. Args: dim (int): The dimension of the embedding. max_position_embeddings (int, opt
eagle/model/modeling_llama_kv.py:135
↓ 1 callersClassLlamaRotaryEmbedding_L31
eagle/traineagle3/modeling_llama_kv.py:208
↓ 1 callersClassLlamaRotaryEmbedding_L31
eagle/model/modeling_llama_kv.py:208
↓ 1 callersClassMixtralBLockSparseTop2MLP
eagle/model/modeling_mixtral_kv.py:426
↓ 1 callersClassMixtralDecoderLayer
eagle/model/modeling_mixtral_kv.py:519
↓ 1 callersClassMixtralRotaryEmbedding
eagle/model/modeling_mixtral_kv.py:207
↓ 1 callersClassMixtralSparseMoeBlock
This implementation is strictly equivalent to standard MoE with full capacity (no dropped tokens). It's faster since it formulates MoE op
eagle/model/modeling_mixtral_kv.py:449
↓ 1 callersClassQwen2DecoderLayer
eagle/model/modeling_qwen2_kv.py:599
↓ 1 callersClassQwen2MLP
eagle/model/modeling_qwen2_kv.py:208
↓ 1 callersClassQwen3Attention
Multi-headed attention from 'Attention Is All You Need' paper
eagle/model/modeling_qwen3_kv.py:224
↓ 1 callersClassQwen3DecoderLayer
eagle/model/modeling_qwen3_kv.py:301
↓ 1 callersClassQwen3MLP
eagle/model/modeling_qwen3_kv.py:136
↓ 1 callersClassQwen3RotaryEmbedding
eagle/model/modeling_qwen3_kv.py:386
↓ 1 callersClassTimer
eagle/testbug/model/ea_modelbs.py:18
↓ 1 callersClassTimer
eagle/model/utils.py:24
↓ 1 callersClassTree
eagle/modeling_eagle.py:584
↓ 1 callersClassTree
eagle/testbug/model/utils_c.py:54
↓ 1 callersClassTree
eagle/model/utils_c.py:57
↓ 1 callersClassdmodel
eagle/testbug/model/ea_model.py:19
ClassEAGLE
eagle/modeling_eagle.py:1555
ClassEAGLE_Config
r""" Copyed from LlamaConfig, the structure of EAGLE consists of a single LlamaDecoder layer. Configuration objects inherit from [`Pretrai
eagle/modeling_eagle.py:175
ClassEConfig
r""" This is the configuration class to store the configuration of a [`LlamaModel`]. It is used to instantiate an LLaMA model according to t
eagle/testbug/model/configs.py:2
ClassEConfig
r""" This is the configuration class to store the configuration of a [`LlamaModel`]. It is used to instantiate an LLaMA model according to t
eagle/traineagle3/configs.py:2
ClassEConfig
r""" This is the configuration class to store the configuration of a [`LlamaModel`]. It is used to instantiate an LLaMA model according to t
eagle/model/configs.py:2
ClassEaModel
eagle/testbug/model/ea_modelbs.py:33
next →1–100 of 129, ranked by callers