MCPcopy Create free account

hub / github.com/jingyaogong/minimind / functions

Functions142 in github.com/jingyaogong/minimind

Method__init__
(self, base_url: str, model_path: str, shared_ckpt_path: str = "./sglang_ckpt", timeout: int = 120)
trainer/rollout_engine.py:100
Method__init__
(self, sampler, batch_size, skip_batches=0)
trainer/trainer_utils.py:135
Method__init__
(self, model_path, device="cuda", dtype=torch.float16)
trainer/trainer_utils.py:161
Method__iter__
(self)
trainer/trainer_utils.py:140
Method__len__
(self)
dataset/lm_dataset.py:44
Method__len__
(self)
dataset/lm_dataset.py:68
Method__len__
(self)
dataset/lm_dataset.py:132
Method__len__
(self)
dataset/lm_dataset.py:205
Method__len__
(self)
dataset/lm_dataset.py:236
Method__len__
(self)
trainer/trainer_utils.py:155
Function_generate
()
scripts/serve_openai_api.py:113
Functionchat_completions
(request: ChatRequest)
scripts/serve_openai_api.py:179
Functionclear_chat_messages
()
scripts/web_demo.py:212
Functioncollate_fn
(batch)
trainer/train_agent.py:456
Functionconvert_jinja_to_json
(jinja_path)
scripts/convert_model.py:115
Functionconvert_json_to_jinja
(json_file_path, output_path)
scripts/convert_model.py:121
Functionconvert_merge_base_lora
(base_torch_path, lora_path, merged_torch_path)
scripts/convert_model.py:105
Functionconvert_torch2transformers_minimind
(torch_path, transformers_path, dtype=torch.float16)
scripts/convert_model.py:16
Functionconvert_transformers2torch
(transformers_path, torch_path)
scripts/convert_model.py:99
Methodflush_cache
(self)
trainer/rollout_engine.py:196
Functionformat_think
(match)
scripts/web_demo.py:174
Functionformat_think_in_progress
(match)
scripts/web_demo.py:182
Functionformat_think_no_start
(match)
scripts/web_demo.py:188
Functionformat_tool_call
(match)
scripts/web_demo.py:152
Methodforward
(self, x)
model/model_lora.py:17
Methodforward
(self, x)
model/model_minimind.py:59
Methodforward
(self, x, position_embeddings, past_key_value=None, use_cache=False, attention_mask=None)
model/model_minimind.py:111
Methodforward
(self, x)
model/model_minimind.py:145
Methodforward
(self, x)
model/model_minimind.py:156
Methodforward
(self, hidden_states, position_embeddings, past_key_value=None, use_cache=False, attention_mask=None)
model/model_minimind.py:186
Methodforward
(self, input_ids, attention_mask=None, past_key_values=None, use_cache=False, **kwargs)
model/model_minimind.py:209
Methodforward
(self, input_ids=None, attention_mask=None, **kwargs)
trainer/train_ppo.py:43
Functionforward_with_lora
(x, layer1=original_forward, layer2=lora)
model/model_lora.py:29
Methodhealth
(self)
trainer/rollout_engine.py:200
Functioninit_chat_messages
()
scripts/web_demo.py:217
Methodon_finalized_text
(self, text: str, stream_end: bool = False)
scripts/serve_openai_api.py:77
Methodpad_to_tensor
(seqs, max_len, pad_val=0)
trainer/rollout_engine.py:162
Functionregenerate_answer
(index)
scripts/web_demo.py:233
Methodrollout
(self, prompt_ids: Tensor, attention_mask: Tensor, num_generations: int, max_new_tokens: int, temperature: flo
trainer/rollout_engine.py:71
Methodrollout
(self, prompt_ids: Tensor, attention_mask: Tensor, num_generations: int, max_new_tokens: int, temperature: flo
trainer/rollout_engine.py:107
Methodupdate_policy
(self, model: torch.nn.Module)
trainer/rollout_engine.py:94
Methodupdate_policy
(self, model: torch.nn.Module)
trainer/rollout_engine.py:175
← previous101–142 of 142, ranked by callers