MCPcopy Create free account

hub / github.com/SafeAILab/EAGLE / functions

Functions596 in github.com/SafeAILab/EAGLE

⨍ Functions 596◇ Types & classes 129↳ Endpoints 4
↓ 1 callersFunctionreorg_answer_file
Sort by question id and de-duplication
eagle/evaluation/gen_ea_answer_llama3chat.py:307
↓ 1 callersFunctionreorg_answer_file
Sort by question id and de-duplication
eagle/evaluation/gen_baseline_answer_vicuna.py:331
↓ 1 callersFunctionreorg_answer_file
Sort by question id and de-duplication
eagle/evaluation/gen_ea_answer_vicuna.py:272
↓ 1 callersFunctionreorg_answer_file
Sort by question id and de-duplication
eagle/evaluation/gen_ea_answer_llama2chat.py:270
↓ 1 callersMethodrepeat_hidden
(self, hidden_state, repeat_num)
eagle/modeling_eagle.py:836
↓ 1 callersMethodreset
(self)
eagle/modeling_eagle.py:711
↓ 1 callersMethodreset
(self)
eagle/testbug/model/cnets.py:468
↓ 1 callersMethodreset
(self)
eagle/model/cnets.py:551
↓ 1 callersMethodreset
(self)
eagle/model/cnets1.py:535
↓ 1 callersMethodreset_kv
(self)
eagle/modeling_eagle.py:832
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_alpha_llama2chat.py:117
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_answer_mix.py:25
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_baseline_answer_mix.py:87
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_answer_ds.py:32
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_begin, question_en
eagle/evaluation/gen_ea_alpha_vicuna.py:122
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_baseline_answer_llama3chat.py:32
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_answer_qwen3.py:30
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_baseline_answer_llama2chat.py:87
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_answer_llama3chat.py:32
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_baseline_answer_vicuna.py:87
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_answer_vicuna.py:30
↓ 1 callersFunctionrun_eval
( base_model_path, ea_model_path, model_id, question_file, question_be
eagle/evaluation/gen_ea_answer_llama2chat.py:25
↓ 1 callersMethodscandata
(self, datapath, tokenizerpath)
eagle/traineagle3/cnets.py:536
↓ 1 callersFunctiontop_accuracy
Computes the accuracy over the k top predictions for the specified values of k
eagle/train/main_deepspeed.py:205
↓ 1 callersFunctiontree_decoding
( model, tree_candidates, past_key_values, tree_position_ids, inp
eagle/modeling_eagle.py:1321
↓ 1 callersFunctionupdate_inference_inputs
( input_ids, attention_mask, candidates, best_candidate, accept_l
eagle/modeling_eagle.py:1463
↓ 1 callersFunctionwarmup
(model)
eagle/application/webui.py:77
Method__call__
(self, features: List[Dict[str, Any]])
eagle/traineagle3/main.py:189
Method__call__
(self, data)
eagle/train/main_deepspeed.py:105
Method__call__
(self, data)
eagle/train/main_deepspeed.py:117
Method__call__
(self, features: List[Dict[str, Any]])
eagle/train/main_deepspeed.py:184
Method__call__
(self, data)
eagle/train/main.py:114
Method__call__
(self, data)
eagle/train/main.py:126
Method__call__
(self, features: List[Dict[str, Any]])
eagle/train/main.py:193
Method__enter__
(self)
eagle/testbug/model/ea_modelbs.py:21
Method__enter__
(self)
eagle/model/utils.py:27
Method__exit__
(self, exc_type, exc_value, traceback)
eagle/testbug/model/ea_modelbs.py:26
Method__exit__
(self, exc_type, exc_value, traceback)
eagle/model/utils.py:32
Method__getitem__
(self, index)
eagle/train/main_deepspeed.py:133
Method__getitem__
(self, index)
eagle/train/main.py:142
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/modeling_eagle.py:95
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/modeling_eagle.py:133
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/modeling_eagle.py:152
Method__init__
( self, vocab_size=32000, hidden_size=4096, intermediate_s
eagle/modeling_eagle.py:237
Method__init__
(self, config)
eagle/modeling_eagle.py:311
Method__init__
(self, config)
eagle/modeling_eagle.py:448
Method__init__
(self, hidden_size, eps=1e-6)
eagle/modeling_eagle.py:482
Method__init__
(self, config, index)
eagle/modeling_eagle.py:496
Method__init__
(self, tree_list)
eagle/modeling_eagle.py:585
Method__init__
(self, config, bias=True)
eagle/modeling_eagle.py:696
Method__init__
Args: base_model: Huggingface causal LLM eaglepath: Path of EAGLE use_tree_attn: Whether to use tre
eagle/modeling_eagle.py:1556
Method__init__
( self, base_model, base_model_name_or_path, ea_model_path,
eagle/testbug/model/ea_modelbs.py:35
Method__init__
(self,parent=None,value=None,dict_key=None)
eagle/testbug/model/utils_c.py:33
Method__init__
(self,tree_list)
eagle/testbug/model/utils_c.py:55
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/testbug/model/cnets.py:112
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/testbug/model/cnets.py:149
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/testbug/model/cnets.py:168
Method__init__
(self, config)
eagle/testbug/model/cnets.py:193
Method__init__
(self, config)
eagle/testbug/model/cnets.py:330
Method__init__
LlamaRMSNorm is equivalent to T5LayerNorm
eagle/testbug/model/cnets.py:363
Method__init__
(self, config,index)
eagle/testbug/model/cnets.py:379
Method__init__
(self,d)
eagle/testbug/model/cnets.py:455
Method__init__
(self,ins=6566,outs=32000)
eagle/testbug/model/cnets.py:682
Method__init__
( self, based, draftd, )
eagle/testbug/model/ea_model.py:30
Method__init__
( self, vocab_size=32000, hidden_size=4096, intermediate_size=11008,
eagle/testbug/model/configs.py:77
Method__init__
Initialize the KVCache. Args: data (torch.Tensor): Initial tensor to store the keys and values. current_leng
eagle/testbug/model/kv_cache.py:17
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/traineagle3/cnets.py:102
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/traineagle3/cnets.py:140
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/traineagle3/cnets.py:159
Method__init__
(self, config)
eagle/traineagle3/cnets.py:186
Method__init__
(self, config, last=True)
eagle/traineagle3/cnets.py:318
Method__init__
LlamaRMSNorm is equivalent to T5LayerNorm
eagle/traineagle3/cnets.py:356
Method__init__
(self, config, last=True)
eagle/traineagle3/cnets.py:373
Method__init__
( self, vocab_size=32000, hidden_size=4096, intermediate_size=11008,
eagle/traineagle3/configs.py:77
Method__init__
(self, hidden_size, eps=1e-6)
eagle/traineagle3/modeling_llama_kv.py:113
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/traineagle3/modeling_llama_kv.py:146
Method__init__
( self, dim=None, max_position_embeddings=2048, base=10000, devic
eagle/traineagle3/modeling_llama_kv.py:209
Method__init__
( self, dim, max_position_embeddings=2048, base=10000,
eagle/traineagle3/modeling_llama_kv.py:308
Method__init__
Initialize the LlamaDynamicNTKScalingRotaryEmbedding. Args: dim (int): The dimensionality of the embedding.
eagle/traineagle3/modeling_llama_kv.py:352
Method__init__
(self, config: LlamaConfig)
eagle/traineagle3/modeling_llama_kv.py:577
Method__init__
(self, config: LlamaConfig)
eagle/traineagle3/modeling_llama_kv.py:791
Method__init__
(self, config: LlamaConfig)
eagle/traineagle3/modeling_llama_kv.py:986
Method__init__
(self, config)
eagle/traineagle3/modeling_llama_kv.py:1207
Method__init__
(self, config)
eagle/traineagle3/modeling_llama_kv.py:1408
Method__init__
(self,parent=None,value=None,dict_key=None)
eagle/model/utils_c.py:36
Method__init__
(self,tree_list)
eagle/model/utils_c.py:58
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/model/cnets.py:111
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/model/cnets.py:149
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
eagle/model/cnets.py:168
Method__init__
(self, config)
eagle/model/cnets.py:194
Method__init__
(self, config)
eagle/model/cnets.py:337
Method__init__
LlamaRMSNorm is equivalent to T5LayerNorm
eagle/model/cnets.py:371
Method__init__
(self, config, last=True)
eagle/model/cnets.py:388
Method__init__
MixtralRMSNorm is equivalent to T5LayerNorm
eagle/model/modeling_mixtral_kv.py:190
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None)
eagle/model/modeling_mixtral_kv.py:208
Method__init__
(self, config: MixtralConfig, layer_idx: Optional[int] = None)
eagle/model/modeling_mixtral_kv.py:300
Method__init__
(self, config: MixtralConfig)
eagle/model/modeling_mixtral_kv.py:427
Method__init__
(self, config)
eagle/model/modeling_mixtral_kv.py:461
Method__init__
(self, config: MixtralConfig, layer_idx: int)
eagle/model/modeling_mixtral_kv.py:520
Method__init__
(self, config)
eagle/model/modeling_mixtral_kv.py:937
← previousnext →201–300 of 596, ranked by callers