MCPcopy Create free account

hub / github.com/SafeAILab/EAGLE / functions

Functions596 in github.com/SafeAILab/EAGLE

Methodget_input_embeddings
(self)
eagle/traineagle3/modeling_llama_kv.py:1217
Methodget_input_embeddings
(self)
eagle/traineagle3/modeling_llama_kv.py:1417
Methodget_input_embeddings
(self)
eagle/model/modeling_mixtral_kv.py:736
Methodget_input_embeddings
(self)
eagle/model/modeling_mixtral_kv.py:948
Methodget_input_embeddings
(self)
eagle/model/modeling_mixtral_kv.py:1105
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen2_kv.py:833
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen2_kv.py:1111
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen2_kv.py:1333
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen2_kv.py:1456
Methodget_input_embeddings
(self)
eagle/model/modeling_llama_kv.py:1003
Methodget_input_embeddings
(self)
eagle/model/modeling_llama_kv.py:1216
Methodget_input_embeddings
(self)
eagle/model/modeling_llama_kv.py:1416
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen3_kv.py:439
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen3_kv.py:637
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen3_kv.py:755
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen3_kv.py:849
Methodget_input_embeddings
(self)
eagle/model/modeling_qwen3_kv.py:914
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_alpha_llama2chat.py:175
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_answer_mix.py:83
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_baseline_answer_mix.py:145
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_answer_ds.py:90
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file, max_new_token,
eagle/evaluation/gen_ea_alpha_vicuna.py:180
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_baseline_answer_llama3chat.py:90
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_answer_qwen3.py:88
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_baseline_answer_llama2chat.py:145
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_answer_llama3chat.py:90
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_baseline_answer_vicuna.py:145
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_answer_vicuna.py:88
Functionget_model_answers
( base_model_path, ea_model_path, model_id, questions, answer_file,
eagle/evaluation/gen_ea_answer_llama2chat.py:83
Methodget_output_embeddings
(self)
eagle/traineagle3/modeling_llama_kv.py:1223
Methodget_output_embeddings
(self)
eagle/model/modeling_mixtral_kv.py:954
Methodget_output_embeddings
(self)
eagle/model/modeling_qwen2_kv.py:1117
Methodget_output_embeddings
(self)
eagle/model/modeling_llama_kv.py:1222
Methodget_output_embeddings
(self)
eagle/model/modeling_qwen3_kv.py:643
Methodget_tokenizer
Get the tokenizer of the base model. Returns: Tokenizer: The tokenizer of the base model.
eagle/testbug/model/ea_modelbs.py:71
Methodget_tokenizer
Get the tokenizer of the base model. Returns: Tokenizer: The tokenizer of the base model.
eagle/testbug/model/ea_model.py:41
Methodinit_tree
(self)
eagle/model/cnets1.py:530
Functioninitialize_tree0
(input_ids, model, past_key_values, logits_processor)
eagle/model/utils.py:210
Functionlen_list
(x,n)
eagle/testbug/model/cnets.py:451
Functionlen_list
(x, n)
eagle/model/cnets.py:474
Functionlen_list
(x, n)
eagle/model/cnets1.py:468
Methodnaive_generate
( self, input_ids, temperature=0.0, top_p=0.0, top
eagle/testbug/model/ea_model.py:202
Methodnaivegenerate
( self, input_ids, attention_mask=None, temperature=0.0,
eagle/testbug/model/ea_modelbs.py:305
Functionpad_path
Pad the given path list with a specific value up to a specified length. Parameters: - path (list): The original list that needs padding.
eagle/testbug/model/utils_c.py:6
Functionpad_path
Pad the given path list with a specific value up to a specified length. Parameters: - path (list): The original list that needs padding.
eagle/model/utils_c.py:9
Functionpadding
(tensor, left=True)
eagle/model/cnets.py:464
Methodpaddingtensor
(self, intensors, N)
eagle/traineagle3/main.py:176
Methodprepare_inputs_for_generation
( self, input_ids, past_key_values=None, attention_mask=No
eagle/traineagle3/modeling_llama_kv.py:1344
Methodprepare_inputs_for_generation
( self, input_ids, past_key_values=None, attention_mask=None,
eagle/model/modeling_qwen2_kv.py:1233
Methodprepare_inputs_for_generation
( self, input_ids, past_key_values=None, attention_mask=None,
eagle/model/modeling_llama_kv.py:1343
Functionpreprocess_function
(examples)
eagle/traineagle3/main.py:66
Methodpreprocess_function
(examples)
eagle/traineagle3/cnets.py:547
Functionprocess_data
(data_chunk)
eagle/traineagle3/cnets.py:457
Methodreduce_kv
(self,kv,numr)
eagle/testbug/model/cnets.py:534
Functionregenerate
(history,session_state)
eagle/application/webui.py:218
Methodrepeat_hidden
(self,hidden_state,repeat_num)
eagle/testbug/model/cnets.py:545
Methodrepeat_kv
(self,kv,numr)
eagle/testbug/model/cnets.py:527
Methodreset_kv
(self)
eagle/model/cnets1.py:669
Functionreset_past_key_values
Resets the current lengths in the passed key-values to zero. This function is designed to be used during the evaluation of a baseline model.
eagle/testbug/model/utils_c.py:179
Functionreset_past_key_values
Resets the current lengths in the passed key-values to zero. This function is designed to be used during the evaluation of a baseline model.
eagle/testbug/model/utils.py:242
Functionreset_past_key_values
Resets the current lengths in the passed key-values to zero. This function is designed to be used during the evaluation of a baseline model.
eagle/model/utils_c.py:182
Functionreset_past_key_values
Resets the current lengths in the passed key-values to zero. This function is designed to be used during the evaluation of a baseline model.
eagle/model/utils.py:264
Methodset_decoder
(self, decoder)
eagle/traineagle3/modeling_llama_kv.py:1229
Methodset_decoder
(self, decoder)
eagle/model/modeling_mixtral_kv.py:960
Methodset_decoder
(self, decoder)
eagle/model/modeling_qwen2_kv.py:1123
Methodset_decoder
(self, decoder)
eagle/model/modeling_llama_kv.py:1228
Methodset_decoder
(self, decoder)
eagle/model/modeling_qwen3_kv.py:649
Methodset_input_embeddings
(self, value)
eagle/traineagle3/modeling_llama_kv.py:1006
Methodset_input_embeddings
(self, value)
eagle/traineagle3/modeling_llama_kv.py:1220
Methodset_input_embeddings
(self, value)
eagle/traineagle3/modeling_llama_kv.py:1420
Methodset_input_embeddings
(self, value)
eagle/model/modeling_mixtral_kv.py:739
Methodset_input_embeddings
(self, value)
eagle/model/modeling_mixtral_kv.py:951
Methodset_input_embeddings
(self, value)
eagle/model/modeling_mixtral_kv.py:1108
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen2_kv.py:836
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen2_kv.py:1114
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen2_kv.py:1336
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen2_kv.py:1459
Methodset_input_embeddings
(self, value)
eagle/model/modeling_llama_kv.py:1006
Methodset_input_embeddings
(self, value)
eagle/model/modeling_llama_kv.py:1219
Methodset_input_embeddings
(self, value)
eagle/model/modeling_llama_kv.py:1419
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen3_kv.py:442
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen3_kv.py:640
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen3_kv.py:758
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen3_kv.py:852
Methodset_input_embeddings
(self, value)
eagle/model/modeling_qwen3_kv.py:917
Methodset_output_embeddings
(self, new_embeddings)
eagle/traineagle3/modeling_llama_kv.py:1226
Methodset_output_embeddings
(self, new_embeddings)
eagle/model/modeling_mixtral_kv.py:957
Methodset_output_embeddings
(self, new_embeddings)
eagle/model/modeling_qwen2_kv.py:1120
Methodset_output_embeddings
(self, new_embeddings)
eagle/model/modeling_llama_kv.py:1225
Methodset_output_embeddings
(self, new_embeddings)
eagle/model/modeling_qwen3_kv.py:646
Methodshape
Return the shape of the data tensor with updated length.
eagle/testbug/model/kv_cache.py:29
Methodshape
Return the shape of the data tensor with updated length.
eagle/model/kv_cache.py:29
Functiontimer
(func)
eagle/testbug/model/utils.py:21
MethodtopK_genrate
(self, hidden_states, input_ids, head, logits_processor)
eagle/model/cnets1.py:673
Functionuser
(user_message, history,session_state)
eagle/application/webui.py:209
Functionwrapper
(*args, **kwargs)
eagle/testbug/model/utils.py:22
← previous501–596 of 596, ranked by callers