MCPcopy Create free account

hub / github.com/Chen-GX/C-3PO / functions

Functions251 in github.com/Chen-GX/C-3PO

⨍ Functions 251◇ Types & classes 36↳ Endpoints 1
Functionf1_score
(prediction, ground_truths)
deploy_servers/retrieve_server/retrieve_code/src/evaluation.py:153
Methodforward
(ctx, x: torch.tensor)
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:9
Methodforward
(self, q_tokens, q_mask, k_tokens, k_mask, stats_prefix="", iter_stats={}, **kwargs)
deploy_servers/retrieve_server/retrieve_code/src/inbatch.py:59
Methodforward
( self, input_ids=None, attention_mask=None, token_type_ids=None, posi
deploy_servers/retrieve_server/retrieve_code/src/contriever.py:64
Methodforward
(self, q_tokens, q_mask, k_tokens, k_mask, stats_prefix="", iter_stats={}, **kwargs)
deploy_servers/retrieve_server/retrieve_code/src/moco.py:104
Functiongather
(x: torch.tensor)
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:21
Functiongather_nograd
(x: torch.tensor)
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:30
Methodgenerate
(self, query_list)
C-3PO/construct_cache.py:29
Methodgenerate
(self, prompt_dict: List[dict])
C-3PO/llm_server.py:130
Methodgenerate_offset
(self)
deploy_servers/retrieve_server/retrieve_code/src/data.py:71
Methodget_encoder
(self)
deploy_servers/retrieve_server/retrieve_code/src/inbatch.py:56
Methodget_encoder
(self, return_encoder_k=False)
deploy_servers/retrieve_server/retrieve_code/src/moco.py:68
Functionget_parameters
(net, verbose=False)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:145
Functionget_varsize
gather tensors of different sizes along the first dimension
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:64
Functionget_world_size
()
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:89
Functioninit_distributed_mode
Handle single and multi-GPU / multi-node / SLURM jobs. Initialize the following variables: - local_rank - global_rank
deploy_servers/retrieve_server/retrieve_code/src/slurm.py:43
Functioninit_logger
(args, stdout_only=False)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:18
Functioninit_signal_handler
Handle signals sent by SLURM for time limit / pre-emption.
deploy_servers/retrieve_server/retrieve_code/src/slurm.py:35
Functioninit_tb_logger
(output_dir)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:201
Methodllm_generate
(self, query_list, max_retries=1e8, retry_delay=1)
C-3PO/llm_server.py:112
Methodllm_planning_role
(self, nodes: List[BaseNode])
C-3PO/search/LLM_planning_role.py:16
Methodllm_query_role
(self, nodes: List[BaseNode])
C-3PO/search/LLM_query_role.py:11
Functionload
(model_class, dir_path, opt, reset_params=False)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:66
Functionload_corpus
(path)
deploy_servers/retrieve_server/retrieve_code/src/data.py:227
Functionload_data
(data_path)
deploy_servers/retrieve_server/retrieve_code/passage_retrieval.py:241
Functionload_data
(opt, tokenizer)
deploy_servers/retrieve_server/retrieve_code/src/data.py:20
Functionload_hf
(object_class, model_name)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:193
Functionload_retriever
(model_path, pooling="average", random_init=False)
deploy_servers/retrieve_server/retrieve_code/src/contriever.py:103
Methodlr_lambda
(self, step)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:98
Methodlr_lambda
(self, step)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:115
Methodmake_decision_role
(self, nodes: List[BaseNode])
C-3PO/search/make_decision_role.py:13
Functionnormalize
(text)
deploy_servers/retrieve_server/retrieve_code/src/normalize_text.py:133
Methodparse
(self)
deploy_servers/retrieve_server/retrieve_code/src/options.py:129
Methodprint_options
(self, opt)
deploy_servers/retrieve_server/retrieve_code/src/options.py:112
Functionprint_tree
(all_save_tree_info)
C-3PO/utils.py:167
Functionread_and_check_ip
(file_path)
C-3PO/llm_server.py:25
Methodreset
(self)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:173
Methodretrieve
(self, messages: List[str], retrieve_times: List[int])
C-3PO/retrieve/retrieve_engine.py:16
Functionsave
(model, optimizer, scheduler, step, opt, dir_path, name)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:46
Functionsearch
()
deploy_servers/retrieve_server/retrieve_code/wiki18/wiki18_serve.py:35
Methodsearch_document_demo
(self, query, n_docs=10)
deploy_servers/retrieve_server/retrieve_code/passage_retrieval.py:197
Methodsetup_retriever_demo
(self, retrieve_model, corpus, corpus_embeddings, n_docs=5, save_or_load_index=False)
deploy_servers/retrieve_server/retrieve_code/passage_retrieval.py:207
Methodsglang_generate
(self, query_list)
C-3PO/proxy.py:54
Functionsig_handler
(signum, frame)
deploy_servers/retrieve_server/retrieve_code/src/slurm.py:18
Methodsingle_query
(self, payload)
C-3PO/retrieve/search_engine.py:47
Methodstats
(self)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:166
Functionstr2bool
(v)
C-3PO/arguments.py:10
Functionterm_handler
(signum, frame)
deploy_servers/retrieve_server/retrieve_code/src/slurm.py:30
Methodtuple_stats
(self)
deploy_servers/retrieve_server/retrieve_code/src/utils.py:170
Functionvarsize_gather_nograd
gather tensors of different sizes along the first dimension
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:41
Functionweighted_average
(x, count)
deploy_servers/retrieve_server/retrieve_code/src/dist_utils.py:119
← previous201–251 of 251, ranked by callers