MCPcopy Create free account

hub / github.com/SafeAILab/EAGLE / functions

Functions596 in github.com/SafeAILab/EAGLE

⨍ Functions 596◇ Types & classes 129↳ Endpoints 4
Method__init__
(self, config)
eagle/model/modeling_mixtral_kv.py:1096
Method__init__
( self, use_eagle3, base_model, base_model_name_or_path,
eagle/model/ea_model.py:27
Method__init__
( self, vocab_size=32000, hidden_size=4096, intermediate_size=11008,
eagle/model/configs.py:77
Method__init__
(self,name)
eagle/model/utils.py:25
Method__init__
Qwen2RMSNorm is equivalent to T5LayerNorm
eagle/model/modeling_qwen2_kv.py:64
Method__init__
( self, dim=None, max_position_embeddings=2048, base=10000,
eagle/model/modeling_qwen2_kv.py:85
Method__init__
(self, config: Qwen2Config, layer_idx: Optional[int] = None)
eagle/model/modeling_qwen2_kv.py:241
Method__init__
(self, *args, **kwargs)
eagle/model/modeling_qwen2_kv.py:357
Method__init__
(self, config: Qwen2Config, layer_idx: int)
eagle/model/modeling_qwen2_kv.py:600
Method__init__
(self, config: Qwen2Config)
eagle/model/modeling_qwen2_kv.py:816
Method__init__
(self, config)
eagle/model/modeling_qwen2_kv.py:1102
Method__init__
(self, config)
eagle/model/modeling_qwen2_kv.py:1324
Method__init__
(self, config)
eagle/model/modeling_qwen2_kv.py:1440
Method__init__
Initialize the KVCache. Args: data (torch.Tensor): Initial tensor to store the keys and values. current_leng
eagle/model/kv_cache.py:17
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/model/cnets1.py:111
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/model/cnets1.py:149
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/model/cnets1.py:168
Method__init__
(self, config)
eagle/model/cnets1.py:194
Method__init__
(self, config)
eagle/model/cnets1.py:342
Method__init__
LlamaRMSNorm is equivalent to T5LayerNorm
eagle/model/cnets1.py:376
Method__init__
(self, config, index)
eagle/model/cnets1.py:393
Method__init__
(self, config, load_emb=False, path=None, bias=True, total_tokens=63, depth=5, top_k=8, threshold=1.0)
eagle/model/cnets1.py:473
Method__init__
(self, hidden_size, eps=1e-6)
eagle/model/modeling_llama_kv.py:113
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/model/modeling_llama_kv.py:146
Method__init__
( self, dim=None, max_position_embeddings=2048, base=10000, device=Non
eagle/model/modeling_llama_kv.py:209
Method__init__
( self, dim, max_position_embeddings=2048, base=10000,
eagle/model/modeling_llama_kv.py:308
Method__init__
Initialize the LlamaDynamicNTKScalingRotaryEmbedding. Args: dim (int): The dimensionality of the embedding.
eagle/model/modeling_llama_kv.py:352
Method__init__
(self, config: LlamaConfig)
eagle/model/modeling_llama_kv.py:577
Method__init__
(self, config: LlamaConfig)
eagle/model/modeling_llama_kv.py:791
Method__init__
(self, config: LlamaConfig)
eagle/model/modeling_llama_kv.py:986
Method__init__
(self, config)
eagle/model/modeling_llama_kv.py:1206
Method__init__
(self, config)
eagle/model/modeling_llama_kv.py:1407
Method__init__
Qwen3RMSNorm is equivalent to T5LayerNorm
eagle/model/modeling_qwen3_kv.py:117
Method__init__
(self, config: Qwen3Config, layer_idx: int)
eagle/model/modeling_qwen3_kv.py:227
Method__init__
(self, config: Qwen3Config, layer_idx: int)
eagle/model/modeling_qwen3_kv.py:302
Method__init__
(self, config: Qwen3Config, device=None)
eagle/model/modeling_qwen3_kv.py:387
Method__init__
(self, config: Qwen3Config)
eagle/model/modeling_qwen3_kv.py:422
Method__init__
(self, config)
eagle/model/modeling_qwen3_kv.py:628
Method__init__
(self, config)
eagle/model/modeling_qwen3_kv.py:746
Method__init__
(self, config)
eagle/model/modeling_qwen3_kv.py:833
Method__init__
(self, config)
eagle/model/modeling_qwen3_kv.py:906
Method__init__
(self, mean=0.0, std=0.0)
eagle/train/main_deepspeed.py:101
Method__init__
(self, std=0.0)
eagle/train/main_deepspeed.py:114
Method__init__
(self, datapath, transform=None)
eagle/train/main_deepspeed.py:126
Method__init__
(self, mean=0.0, std=0.0)
eagle/train/main.py:110
Method__init__
(self, std=0.0)
eagle/train/main.py:123
Method__init__
(self, datapath, transform=None)
eagle/train/main.py:135
Method__len__
(self)
eagle/train/main_deepspeed.py:130
Method__len__
(self)
eagle/train/main.py:139
Function_get_unpad_data
(attention_mask)
eagle/model/modeling_mixtral_kv.py:176
Method_init_weights
(self, module)
eagle/traineagle3/modeling_llama_kv.py:894
Method_init_weights
(self, module)
eagle/model/modeling_mixtral_kv.py:628
Method_init_weights
(self, module)
eagle/model/modeling_qwen2_kv.py:717
Method_init_weights
(self, module)
eagle/model/modeling_llama_kv.py:894
Method_init_weights
(self, module)
eagle/model/modeling_qwen3_kv.py:372
Method_prepare_decoder_attention_mask
(self, attention_mask, input_shape, inputs_embeds, past_key_values_length)
eagle/testbug/model/cnets.py:472
Method_reorder_cache
(past_key_values, beam_idx)
eagle/traineagle3/modeling_llama_kv.py:1380
Method_reorder_cache
(past_key_values, beam_idx)
eagle/model/modeling_llama_kv.py:1379
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/modeling_eagle.py:137
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/modeling_eagle.py:156
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/testbug/model/cnets.py:153
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/testbug/model/cnets.py:172
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/traineagle3/cnets.py:144
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/traineagle3/cnets.py:163
Method_set_cos_sin_cache
Set the cosine and sine cache for the rotary embeddings. Args: seq_len (int): The sequence length. devi
eagle/traineagle3/modeling_llama_kv.py:319
Method_set_cos_sin_cache
Set the cached values for cosine and sine. Args: seq_len (int): The sequence length. device: The device
eagle/traineagle3/modeling_llama_kv.py:373
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/model/cnets.py:153
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/model/cnets.py:172
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/model/cnets1.py:153
Method_set_cos_sin_cache
(self, seq_len, device, dtype)
eagle/model/cnets1.py:172
Method_set_cos_sin_cache
Set the cosine and sine cache for the rotary embeddings. Args: seq_len (int): The sequence length. device (s
eagle/model/modeling_llama_kv.py:319
Method_set_cos_sin_cache
Set the cached values for cosine and sine. Args: seq_len (int): The sequence length. device: The device to p
eagle/model/modeling_llama_kv.py:373
Method_set_gradient_checkpointing
(self, module, value=False)
eagle/traineagle3/modeling_llama_kv.py:905
Method_set_gradient_checkpointing
(self, module, value=False)
eagle/model/modeling_llama_kv.py:905
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/modeling_eagle.py:349
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/testbug/model/cnets.py:231
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/traineagle3/cnets.py:224
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/traineagle3/modeling_llama_kv.py:636
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/model/cnets.py:238
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/model/modeling_mixtral_kv.py:337
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/model/cnets1.py:243
Method_shape
(self, tensor: torch.Tensor, seq_len: int, bsz: int)
eagle/model/modeling_llama_kv.py:636
Methodacc
(self,data,head,max_length=5)
eagle/testbug/model/cnets.py:626
Functionbot
(history, temperature, top_p, use_EaInfer, highlight_EaInfer,session_state,)
eagle/application/webui.py:97
Functionclear
(history,session_state)
eagle/application/webui.py:232
Methodcopy
Copy values from the current data at specified indices to a new location. Args: indices (torch.Tensor): Indices of the d
eagle/testbug/model/kv_cache.py:38
Methodcopy
Copy values from the current data at specified indices to a new location. Args: indices (torch.Tensor): Indices of the d
eagle/model/kv_cache.py:38
Functioncount_parameters
(model)
eagle/testbug/model/cnets.py:692
Functioncount_parameters
(model)
eagle/model/cnets.py:835
Functioncount_parameters
(model)
eagle/model/cnets1.py:828
Methodcreate_custom_forward
(module)
eagle/modeling_eagle.py:799
Methodcreate_custom_forward
(module)
eagle/traineagle3/cnets.py:799
Methodcreate_custom_forward
(module)
eagle/traineagle3/modeling_llama_kv.py:1148
Methodcreate_custom_forward
(module)
eagle/model/cnets1.py:636
Methodcreate_custom_forward
(module)
eagle/model/modeling_llama_kv.py:1147
Methodcustom_forward
(*inputs)
eagle/modeling_eagle.py:800
Methodcustom_forward
(*inputs)
eagle/traineagle3/cnets.py:800
Methodcustom_forward
(*inputs)
eagle/traineagle3/modeling_llama_kv.py:1149
Methodcustom_forward
(*inputs)
eagle/model/cnets1.py:637
Methodcustom_forward
(*inputs)
eagle/model/modeling_llama_kv.py:1148
← previousnext →301–400 of 596, ranked by callers