Functions — github.com/Yuan-Li-FNLP/R3-RAG

hub / github.com/Yuan-Li-FNLP/R3-RAG / functions

Functions1,737 in github.com/Yuan-Li-FNLP/R3-RAG

⨍ Functions 1,737◇ Types & classes 293↳ Endpoints 12

↓ 1 callersFunctioncreate_chat_completion_response

( request: "ChatCompletionRequest", chat_model: "ChatModel" )

train/LLaMA-Factory/src/llamafactory/api/chat.py:142

↓ 1 callersFunctioncreate_ds_config

r""" Creates deepspeed config.

train/LLaMA-Factory/src/llamafactory/webui/utils.py:239

↓ 1 callersFunctioncreate_eval_tab

(engine: "Engine")

train/LLaMA-Factory/src/llamafactory/webui/components/eval.py:32

↓ 1 callersFunctioncreate_evaluate

(engine: "Engine")

tool/FlashRAG/webui/components/evaluate.py:8

↓ 1 callersFunctioncreate_export_tab

(engine: "Engine")

train/LLaMA-Factory/src/llamafactory/webui/components/export.py:104

↓ 1 callersFunctioncreate_generator

()

tool/FlashRAG/webui/components/generate_setting.py:8

↓ 1 callersFunctioncreate_infer_tab

(engine: "Engine")

train/LLaMA-Factory/src/llamafactory/webui/components/infer.py:32

↓ 1 callersFunctioncreate_method

()

tool/FlashRAG/webui/components/method_setting.py:9

↓ 1 callersMethodcreate_optimizer

( self, model: "AutoModelForCausalLMWithValueHead", training_args: "Seq2SeqTrainingArg

train/LLaMA-Factory/src/llamafactory/train/ppo/trainer.py:302

↓ 1 callersFunctioncreate_preview

(engine: "Engine")

tool/FlashRAG/webui/components/preview.py:9

↓ 1 callersFunctioncreate_rerank

()

tool/FlashRAG/webui/components/rerank_setting.py:8

↓ 1 callersFunctioncreate_retrieve

()

tool/FlashRAG/webui/components/retrieve_setting.py:9

↓ 1 callersFunctioncreate_reward_model

r""" Creates reward model for PPO training.

train/LLaMA-Factory/src/llamafactory/train/trainer_utils.py:146

↓ 1 callersMethodcreate_scheduler

( self, training_args: "Seq2SeqTrainingArguments", num_training_steps: int, optimizer: "torch.optim.Op

train/LLaMA-Factory/src/llamafactory/train/ppo/trainer.py:329

↓ 1 callersFunctioncreate_score_evaluation_response

( request: "ScoreEvaluationRequest", chat_model: "ChatModel" )

train/LLaMA-Factory/src/llamafactory/api/chat.py:229

↓ 1 callersFunctioncreate_step_html

创建单个步骤的HTML显示

startup/R3-RAG.py:237

↓ 1 callersFunctioncreate_stream_chat_completion_response

( request: "ChatCompletionRequest", chat_model: "ChatModel" )

train/LLaMA-Factory/src/llamafactory/api/chat.py:193

↓ 1 callersFunctioncreate_top

()

train/LLaMA-Factory/src/llamafactory/webui/components/top.py:32

↓ 1 callersFunctioncreate_train_tab

(engine: "Engine")

train/LLaMA-Factory/src/llamafactory/webui/components/train.py:37

↓ 1 callersFunctioncreate_ui

()

tool/FlashRAG/webui/interface.py:14

↓ 1 callersFunctioncreate_visualization_interface

创建Gradio界面

startup/R3-RAG.py:402

↓ 1 callersFunctioncreate_vllm_engines

( num_engines: int, tensor_parallel_size: int, pretrain: str, seed: int, enable_prefix_cac

train/R3RAG_OpenRLHF/openrlhf/trainer/ray/vllm_engine.py:84

↓ 1 callersFunctioncreate_web_demo

()

train/LLaMA-Factory/src/llamafactory/webui/interface.py:72

↓ 1 callersFunctiondump_layernorm

r""" Dumps the layernorm parameters in the model. The model is already unwrapped (and gathered).

train/LLaMA-Factory/src/llamafactory/train/ppo/ppo_utils.py:69

↓ 1 callersMethoddynamic_preprocess

(self, image, min_num=1, max_num=12, image_size=448, use_thumbnail=False)

tool/FlashRAG/flashrag/generator/multimodal_generator.py:190

↓ 1 callersMethodencode

(self, query_list: List[str], batch_size=64, modal="image")

tool/FlashRAG/flashrag/retriever/encoder.py:190

↓ 1 callersMethodencode_all

(self)

tool/FlashRAG/flashrag/retriever/index_builder.py:234

↓ 1 callersMethodencode_all_clip

(self)

tool/FlashRAG/flashrag/retriever/index_builder.py:245

↓ 1 callersMethodencode_multiturn

r""" Returns multiple pairs of token ids representing prompts and responses respectively.

train/LLaMA-Factory/src/llamafactory/data/template.py:73

↓ 1 callersMethodencode_passages

(self, batch_text_passages: List[List[str]])

tool/FlashRAG/flashrag/generator/generator.py:84

↓ 1 callersMethodevaluate

(self, eval_dataloader, steps=0)

train/R3RAG_OpenRLHF/openrlhf/trainer/prm_trainer.py:203

↓ 1 callersMethodevaluate

(self, steps=0)

train/R3RAG_OpenRLHF/openrlhf/trainer/kto_trainer.py:216

↓ 1 callersMethodevaluate

(self, eval_dataloader, steps=0)

train/R3RAG_OpenRLHF/openrlhf/trainer/rm_trainer.py:232

↓ 1 callersMethodevaluate

(self, eval_dataloader, steps=0)

train/R3RAG_OpenRLHF/openrlhf/trainer/dpo_trainer.py:253

↓ 1 callersMethodevaluate

(self, eval_dataloader, steps=0)

train/R3RAG_OpenRLHF/openrlhf/trainer/sft_trainer.py:240

↓ 1 callersFunctionexist_and_not_none

(d, key)

train/R3RAG_OpenRLHF/openrlhf/datasets/utils.py:16

↓ 1 callersMethodextract

r""" Extract a list of tuples from the response message if using tools. Each tuple consists of function name and function arguments.

train/LLaMA-Factory/src/llamafactory/data/formatter.py:39

↓ 1 callersMethodextract_document_triples

Extract triples from documents associated with each query, handling duplicates and generating prompts for LLM processing.

tool/FlashRAG/flashrag/refiner/kg_refiner.py:219

↓ 1 callersMethodextract_judge_score

(answer: str, split_str: str = "Total rating:")

tool/FlashRAG/flashrag/evaluator/metrics.py:546

↓ 1 callersFunctionfactory

(name_or_path)

tool/FlashRAG/flashrag/pipeline/replug_utils.py:232

↓ 1 callersMethodfind_closest_aspect_ratio

(self, aspect_ratio, target_ratios, width, height, image_size)

tool/FlashRAG/flashrag/generator/multimodal_generator.py:175

↓ 1 callersFunctionfind_expanded_modules

r""" Finds the modules in the expanded blocks to apply lora.

train/LLaMA-Factory/src/llamafactory/model/model_utils/misc.py:57

↓ 1 callersFunctionfind_sublist_first_occurrence

(list1, list2, start_index=0)

train/R3RAG_OpenRLHF/test.py:4

↓ 1 callersMethodfit

(self, args, consumed_samples=0, num_update_steps_per_epoch=None)

train/R3RAG_OpenRLHF/openrlhf/trainer/prm_trainer.py:91

↓ 1 callersMethodfit

(self, args, consumed_samples=0, num_update_steps_per_epoch=None)

train/R3RAG_OpenRLHF/openrlhf/trainer/kto_trainer.py:100

↓ 1 callersMethodfit

(self, args, consumed_samples=0, num_update_steps_per_epoch=None)

train/R3RAG_OpenRLHF/openrlhf/trainer/rm_trainer.py:101

↓ 1 callersMethodfit

(self, args, consumed_samples=0, num_update_steps_per_epoch=None)

train/R3RAG_OpenRLHF/openrlhf/trainer/dpo_trainer.py:108

↓ 1 callersMethodfit

( self, args, prompts_dataloader, pretrain_dataloader, consumed_sample

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_trainer.py:189

↓ 1 callersMethodfit

(self, args, consumed_samples=0, num_update_steps_per_epoch=None)

train/R3RAG_OpenRLHF/openrlhf/trainer/sft_trainer.py:103

↓ 1 callersFunctionflatten_dict

( nested_dict: Dict[str, Any], parent_key: str = "", sep: str = "." )

tool/FlashRAG/webui/utils.py:36

↓ 1 callersMethodflush

(self)

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker.py:1181

↓ 1 callersMethodflush

(self)

tool/FlashRAG/webui/utils.py:63

↓ 1 callersFunctionformat

(input_file, output_file)

data/construct/format_single_query/format.py:60

↓ 1 callersMethodformat_evidences

(self, evidences: List[str])

tool/FlashRAG/flashrag/pipeline/active_pipeline.py:1106

↓ 1 callersMethodformat_example

r""" Converts dataset examples to messages.

train/LLaMA-Factory/src/llamafactory/eval/template.py:36

↓ 1 callersMethodformat_reference

(self, retrieval_result)

tool/FlashRAG/flashrag/prompt/base_prompt.py:192

↓ 1 callersMethodformat_reference

(self, retrieval_result)

tool/FlashRAG/flashrag/refiner/refiner.py:115

↓ 1 callersMethodformat_reference

(self, retrieval_result)

tool/FlashRAG/flashrag/refiner/kg_refiner.py:543

↓ 1 callersMethodformat_reference

(self, doc_item)

tool/FlashRAG/flashrag/pipeline/branching_pipeline.py:28

↓ 1 callersMethodformat_reference_chat

(self, retrieval_result)

tool/FlashRAG/webui/chat_pipelines/base_chat_pipeline.py:6

↓ 1 callersMethodfunction_formatter

r""" Generates the assistant message including all the tool calls.

train/LLaMA-Factory/src/llamafactory/data/tool_utils.py:79

↓ 1 callersFunctiongen_cmd

r""" Generates arguments for previewing.

train/LLaMA-Factory/src/llamafactory/webui/utils.py:108

↓ 1 callersFunctiongen_loss_plot

r""" Plots loss curves in LlamaBoard.

train/LLaMA-Factory/src/llamafactory/extras/ploting.py:51

↓ 1 callersFunctiongenerate

(args)

train/R3RAG_OpenRLHF/openrlhf/cli/interactive_chat.py:7

↓ 1 callersFunctiongenerate_naive_generation_cot_prompt

(question)

benchmark/NaiveGeneration/src/inference.py:118

↓ 1 callersFunctiongenerate_naive_generation_prompt

(question)

benchmark/NaiveGeneration/src/inference.py:144

↓ 1 callersFunctiongenerate_naive_rag_cot_prompt

(question, doc)

benchmark/NaiveRAG/src/inference.py:157

↓ 1 callersFunctiongenerate_naive_rag_prompt

(question, doc)

benchmark/NaiveRAG/src/inference.py:146

↓ 1 callersMethodgenerate_samples

Generate samples and return in batches.

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_prm_orm.py:281

↓ 1 callersMethodgenerate_samples

Generate samples and return in batches.

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_orm.py:281

↓ 1 callersMethodgenerate_samples

Generate samples and return in batches.

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_only_prm.py:281

↓ 1 callersMethodgenerate_samples

Generate samples and return in batches.

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker.py:281

↓ 1 callersMethodgenerate_step_samples

(self, all_prompts: List[str], **generate_kwargs)

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_prm_orm.py:308

↓ 1 callersMethodgenerate_step_samples

(self, all_prompts: List[str], **generate_kwargs)

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_orm.py:308

↓ 1 callersMethodgenerate_step_samples

(self, all_prompts: List[str], **generate_kwargs)

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_only_prm.py:308

↓ 1 callersMethodgenerate_step_samples

(self, all_prompts: List[str], **generate_kwargs)

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker.py:308

↓ 1 callersMethodgenerate_tree_of_thoughts_batch

(self, initial_prompts_batch: List[str])

tool/FlashRAG/flashrag/pipeline/active_pipeline.py:1115

↓ 1 callersFunctiongetFixedUrl

(req)

tool/FlashRAG/docs/sw.js:18

↓ 1 callersMethodget_advantages_and_returns

Function that computes advantages and returns from rewards and values. Calculated as in the original PPO paper: https://arxiv.org/abs/1707.063

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_prm_orm.py:771

↓ 1 callersMethodget_advantages_and_returns

Function that computes advantages and returns from rewards and values. Calculated as in the original PPO paper: https://arxiv.org/abs/1707.063

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_orm.py:773

↓ 1 callersMethodget_advantages_and_returns

Function that computes advantages and returns from rewards and values. Calculated as in the original PPO paper: https://arxiv.org/abs/1707.063

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_only_prm.py:771

↓ 1 callersMethodget_advantages_and_returns

Function that computes advantages and returns from rewards and values. Calculated as in the original PPO paper: https://arxiv.org/abs/1707.063

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker.py:771

↓ 1 callersMethodget_answer

(self, question, search_chain_length, split_flag=True)

startup/RRAG.py:237

↓ 1 callersMethodget_answer_token_indices

Obtain the index of token corresponsding to the option

tool/FlashRAG/flashrag/refiner/kg_refiner.py:359

↓ 1 callersMethodget_answer_with_visualization

(self, question, search_chain_length=10)

startup/R3-RAG.py:148

↓ 1 callersFunctionget_args

()

benchmark/NaiveRAG/src/inference.py:13

↓ 1 callersFunctionget_args

()

benchmark/NaiveGeneration/src/inference.py:9

↓ 1 callersFunctionget_args

()

benchmark/R3-RAG/src/inference_4gpu.py:89

↓ 1 callersFunctionget_args

()

benchmark/R3-RAG/src/inference_new.py:89

↓ 1 callersFunctionget_args

()

benchmark/R3-RAG/src/cal_metric.py:257

↓ 1 callersFunctionget_args

()

benchmark/R3-RAG/src/inference.py:89

↓ 1 callersFunctionget_args

()

benchmark/retriever/src/retrive_server.py:12

↓ 1 callersFunctionget_args

()

benchmark/retriever/src/retrive_server_bm25.py:15

↓ 1 callersFunctionget_args

()

benchmark/retriever/src/retrive_server_bge.py:12

↓ 1 callersFunctionget_background_loop

()

tool/FlashRAG/flashrag/generator/openai_generator.py:14

↓ 1 callersMethodget_compressed_input

( self, loss, input_ids, attention_mask, end=200, iterative_si

tool/FlashRAG/flashrag/refiner/llmlingua_compressor.py:1437

↓ 1 callersFunctionget_console_scripts

()

train/LLaMA-Factory/setup.py:37

↓ 1 callersMethodget_cumulative_returns

Function that computes advantages and returns from rewards using REINFORCE. REINFORCE uses cumulative returns without the GAE (Genera

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_prm_orm.py:829

↓ 1 callersMethodget_cumulative_returns

Function that computes advantages and returns from rewards using REINFORCE. REINFORCE uses cumulative returns without the GAE (Genera

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_orm.py:831

↓ 1 callersMethodget_cumulative_returns

Function that computes advantages and returns from rewards using REINFORCE. REINFORCE uses cumulative returns without the GAE (Genera

train/R3RAG_OpenRLHF/openrlhf/trainer/ppo_utils/experience_maker_only_prm.py:829

← previous next →601–700 of 1,737, ranked by callers