MCPcopy Create free account

hub / github.com/XiaoDu-flying/Interlat / functions

Functions1,013 in github.com/XiaoDu-flying/Interlat

↓ 131 callersMethodto
(self, target)
core_training/fastchat/model/rwkv_model.py:27
↓ 114 callersFunctionget_conv_template
Get a conversation template.
core_training/fastchat/conversation.py:427
↓ 90 callersFunctionregister_model_adapter
Register a model adapter.
core_training/fastchat/model/model_adapter.py:135
↓ 86 callersFunctionregister_conv_template
Register a new conversation template.
core_training/fastchat/conversation.py:417
↓ 81 callersFunctionregister_model_info
( full_names: List[str], simple_name: str, link: str, description: str )
core_training/fastchat/model/model_registry.py:12
↓ 43 callersMethodappend_message
Append a new message.
core_training/fastchat/conversation.py:304
↓ 37 callersMethodload
Load config from JSON file
eval/math/config.py:67
↓ 33 callersFunctionget_ip
(request: gr.Request)
core_training/fastchat/serve/gradio_web_server.py:250
↓ 25 callersMethodto_gradio_chatbot
(self)
core_training/fastchat/serve/gradio_web_server.py:105
↓ 22 callersFunctionrank0_print
(*args, **kwargs)
compression_training/compress.py:126
↓ 18 callersFunctioncreate_error_response
(code: int, message: str)
core_training/fastchat/serve/openai_api_server.py:134
↓ 14 callersMethodsave_pretrained
(self, save_directory, **kwargs)
core_training/hidden_model/custom_model.py:784
↓ 13 callersMethodget_prompt
Get the prompt for generation.
core_training/fastchat/conversation.py:69
↓ 13 callersMethodwrite
(self, buf)
core_training/fastchat/utils.py:97
↓ 12 callersMethodload_model
(self, model_path: str, from_pretrained_kwargs: dict)
core_training/fastchat/model/model_adapter.py:1779
↓ 12 callersMethodupdate_last_message
Update the last output. The last message is typically set to be None when constructing the prompt, so we need to update it in-place a
core_training/fastchat/conversation.py:308
↓ 11 callersFunctionbuild_logger
(logger_name, logger_filename)
core_training/fastchat/utils.py:25
↓ 11 callersFunctionfetch_remote
(url, pload=None, name=None)
core_training/fastchat/serve/openai_api_server.py:76
↓ 11 callersMethodsave
Save config to JSON file
eval/math/config.py:60
↓ 10 callersMethoddict
(self)
core_training/fastchat/serve/gradio_web_server.py:108
↓ 10 callersFunctionis_partial_stop
Check whether the output contains a partial stop str.
core_training/fastchat/utils.py:299
↓ 9 callersMethodcreate_custom
Create custom configuration with overrides
data_collection/config.py:112
↓ 9 callersFunctionget_conversation_template
Get the default conversation template.
core_training/fastchat/model/model_adapter.py:381
↓ 8 callersMethodget_input_embeddings
(self)
eval/alfworld/eval_agent/agents/plan_agent.py:59
↓ 8 callersMethodget_input_embeddings
(self)
core_training/hidden_model/custom_model.py:761
↓ 7 callersMethodclose
(self)
eval/alfworld/eval_agent/envs/textcraft_env.py:542
↓ 7 callersMethodget_conv_template
(self)
core_training/fastchat/serve/base_model_worker.py:167
↓ 7 callersMethodget_input_embeddings
(self)
eval/alfworld/eval_agent/agents/model_insert_hidden.py:1504
↓ 7 callersFunctionget_model_adapter
Get a model adapter for a model_path.
core_training/fastchat/model/model_adapter.py:141
↓ 6 callersFunctioncount_parameters
(module)
eval/alfworld/eval_agent/agents/hidden_agent.py:141
↓ 6 callersMethodcount_token
(self, params)
core_training/fastchat/serve/base_model_worker.py:152
↓ 6 callersFunctionget_model_list
(controller_url, register_api_endpoint_file, multimodal)
core_training/fastchat/serve/gradio_web_server.py:131
↓ 6 callersMethodget_queue_length
(self)
core_training/fastchat/serve/base_model_worker.py:131
↓ 6 callersMethodget_status
(self)
core_training/fastchat/serve/base_model_worker.py:145
↓ 6 callersMethodinit_heart_beat
(self)
core_training/fastchat/serve/base_model_worker.py:80
↓ 6 callersMethodnormalize
Normalize answer format
eval/math/math_evaluator.py:106
↓ 6 callersFunctionnow_ms
()
eval/alfworld/eval_agent/main.py:51
↓ 6 callersMethodto_openai_api_messages
Convert the conversation to OpenAI chat completion format.
core_training/fastchat/conversation.py:367
↓ 5 callersFunction_is_local_process_zero
(state)
compression_training/callbacks.py:28
↓ 5 callersMethodgenerate
(self, params)
core_training/fastchat/serve/mlx_worker.py:165
↓ 5 callersFunctionget_context_length
Get the context length of a model from a huggingface model config.
core_training/fastchat/utils.py:332
↓ 5 callersFunctionget_memory_size
计算模块的内存占用(MB)
eval/alfworld/eval_agent/agents/hidden_agent.py:145
↓ 5 callersFunctionget_worker_address
Get worker address based on the requested model :param model_name: The worker's model name :return: Worker address from the controller
core_training/fastchat/serve/openai_api_server.py:368
↓ 5 callersMethodload
Load configuration from JSON file
data_collection/config.py:140
↓ 5 callersMethodmatch
(self, model_path: str)
core_training/fastchat/model/model_adapter.py:1776
↓ 5 callersFunctionprompt_with_icl
(instruction, raw_icl, cur_task, icl_num=2)
eval/alfworld/eval_agent/prompt/templates.py:18
↓ 5 callersFunctionrank0_print
(*args)
core_training/train.py:138
↓ 5 callersMethodset_system_message
Set the system message.
core_training/fastchat/conversation.py:300
↓ 5 callersFunctionvote_last_response
(states, vote_type, model_selectors, request: gr.Request)
core_training/fastchat/serve/gradio_block_arena_named.py:67
↓ 5 callersFunctionvote_last_response
(states, vote_type, model_selectors, request: gr.Request)
core_training/fastchat/serve/gradio_block_arena_anony.py:64
↓ 4 callersMethod_iter_named_params
(self, model)
core_training/callbacks.py:442
↓ 4 callersFunction_write_parquet_shard
Write rows to a single Parquet file
data_collection/math_collection.py:88
↓ 4 callersFunctionadd_model_args
(parser)
core_training/fastchat/model/model_adapter.py:465
↓ 4 callersFunctioncast_sd
(sd)
compression_training/compress.py:193
↓ 4 callersFunctioncheck_model
(request)
core_training/fastchat/serve/openai_api_server.py:145
↓ 4 callersMethodfrom_dict
Create config from dictionary
eval/math/config.py:56
↓ 4 callersFunctionget_conv_log_filename
()
core_training/fastchat/serve/gradio_web_server.py:125
↓ 4 callersMethodget_default_conv_template
(self, model_path: str)
core_training/fastchat/model/model_adapter.py:1796
↓ 4 callersFunctionget_gen_params
( model_name: str, worker_addr: str, messages: Union[str, List[Dict[str, str]]], *, temper
core_training/fastchat/serve/openai_api_server.py:269
↓ 4 callersFunctionget_model_description_md
(models)
core_training/fastchat/serve/gradio_web_server.py:579
↓ 4 callersMethodget_user_call_stats
( self, target_model=None, most_recent_min: int = 60, top_k: int = 20 )
core_training/fastchat/serve/call_monitor.py:122
↓ 4 callersFunctionitem_id_to_str
(item_id: str)
eval/alfworld/eval_agent/envs/textcraft_env.py:66
↓ 4 callersMethoditem_str_to_obj
(self, item)
eval/alfworld/eval_agent/envs/textcraft_env.py:501
↓ 4 callersFunctionrank0_print
(*args)
core_training/data_processor.py:25
↓ 4 callersFunctionreload_conv
Reprints the conversation from the start.
core_training/fastchat/serve/inference.py:400
↓ 3 callersMethod__init__
(self, d_model: int, rank: int = 4, hidden_factor: int = 4)
eval/alfworld/eval_agent/agents/align_block.py:47
↓ 3 callersMethod__init__
(self, hidden_size: int, num_heads: int = 8, dropout: float = 0.1)
compression_training/compress.py:164
↓ 3 callersMethod_first_supervised_pos
(self, labels, attn)
core_training/hidden_model/custom_model.py:637
↓ 3 callersMethod_forward_with_hidden_states_curriculum
( self, input_ids, plan_ids, attention_mask, inputs_embeds, labels, human_end_position
core_training/hidden_model/custom_model.py:318
↓ 3 callersFunctionacquire_worker_semaphore
(worker)
core_training/fastchat/serve/huggingface_api_worker.py:209
↓ 3 callersFunctionacquire_worker_semaphore
()
core_training/fastchat/serve/multi_model_worker.py:76
↓ 3 callersFunctionacquire_worker_semaphore
()
core_training/fastchat/serve/base_model_worker.py:184
↓ 3 callersFunctioncheck_length
(request, prompt, max_tokens, worker_addr)
core_training/fastchat/serve/openai_api_server.py:158
↓ 3 callersFunctioncheck_requests
(request)
core_training/fastchat/serve/openai_api_server.py:183
↓ 3 callersMethodcopy
(self)
core_training/fastchat/conversation.py:388
↓ 3 callersFunctioncount_parameters
(module)
compression_training/compress.py:1040
↓ 3 callersMethodextract_task_from_conversation
Extract the main task description from ALFWorld conversation format Args: conversation_item: ALFWorld conversation traje
data_collection/alfworld_task_collector.py:150
↓ 3 callersFunctionfind_awq_ckpt
(awq_config: AWQConfig)
core_training/fastchat/modules/awq.py:75
↓ 3 callersMethodgenerate
(self, *args, **kwargs)
eval/alfworld/eval_agent/agents/plan_agent.py:81
↓ 3 callersFunctiongenerate_completion
(payload: Dict[str, Any], worker_addr: str)
core_training/fastchat/serve/openai_api_server.py:703
↓ 3 callersMethodgenerate_stream_gate
(self, params)
core_training/fastchat/serve/sglang_worker.py:164
↓ 3 callersMethodget_images
(self)
core_training/fastchat/conversation.py:290
↓ 3 callersMethodget_model_call_stats
( self, target_model=None, most_recent_min: int = 60, top_k: int = 20 )
core_training/fastchat/serve/call_monitor.py:102
↓ 3 callersMethodget_worker_status
(self, worker_name: str)
core_training/fastchat/serve/controller.py:104
↓ 3 callersFunctionload_demo_single
(models, url_params)
core_training/fastchat/serve/gradio_web_server.py:177
↓ 3 callersFunctionload_model
Load a model from Hugging Face.
core_training/fastchat/model/model_adapter.py:184
↓ 3 callersFunctionmoderation_filter
(text, model_list)
core_training/fastchat/utils.py:178
↓ 3 callersFunctionrelease_worker_semaphore
(worker)
core_training/fastchat/serve/huggingface_api_worker.py:205
↓ 3 callersMethodresize_token_embeddings
(self, new_num_tokens, **kwargs)
core_training/hidden_model/custom_model.py:773
↓ 3 callersFunctionstring_args
(args, args_list)
core_training/fastchat/serve/launch_all_serve.py:209
↓ 3 callersFunctionvote_last_response
(state, vote_type, model_selector, request: gr.Request)
core_training/fastchat/serve/gradio_web_server.py:203
↓ 2 callersMethod__process_embed_chunk
(self, input_ids, attention_mask, **model_type_dict)
core_training/fastchat/serve/model_worker.py:151
↓ 2 callersFunction_add_to_set
(s, new_stop)
eval/alfworld/eval_agent/agents/fastchat_agent.py:14
↓ 2 callersFunction_add_to_set
(s, new_stop)
core_training/fastchat/serve/openai_api_server.py:260
↓ 2 callersMethod_assemble_with_latent
(self, inputs_embeds, attention_mask, labels, human_end_positions, latent_list)
compression_training/compress.py:289
↓ 2 callersMethod_collect_target_params
Collect parameters of target layers (custom layers + first layer of base_model)
core_training/callbacks.py:162
↓ 2 callersMethod_forward_with_hidden_states_curriculum
( self, input_ids, plan_ids, attention_mask, inputs_embeds, labels, human_end_position
eval/alfworld/eval_agent/agents/model_insert_hidden.py:1138
↓ 2 callersMethod_get_unwrapped_model
Get the unwrapped model
core_training/callbacks.py:155
↓ 2 callersMethod_init_mha
(mha: nn.MultiheadAttention)
eval/alfworld/eval_agent/agents/align_block.py:206
↓ 2 callersFunction_is_dist_initialized
()
compression_training/callbacks.py:20
next →1–100 of 1,013, ranked by callers