Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Mwie1024/Extra-CoT
/ functions
Functions
1,850 in github.com/Mwie1024/Extra-CoT
⨍
Functions
1,850
◇
Types & classes
174
↳
Endpoints
4
↓ 779 callers
Method
append
r"""Add the user input to chatbot. Inputs: infer.chatbot, infer.messages, infer.role, infer.query, infer.escape_html Output: infer.ch
llamafactory/src/llamafactory/webui/chatter.py:175
↓ 330 callers
Method
join
(self, attr: dict[str, Any])
llamafactory/src/llamafactory/data/parser.py:72
↓ 121 callers
Function
register_model_group
( models: dict[str, dict[DownloadSource, str]], template: Optional[str] = None, multimodal: bool =
llamafactory/src/llamafactory/extras/constants.py:139
↓ 113 callers
Function
register_template
r"""Register a chat template. To add the following chat template: ``` <s><user>user prompt here <model>model response here</s> <u
llamafactory/src/llamafactory/data/template.py:459
↓ 102 callers
Method
info_rank0
(self, *args, **kwargs)
llamafactory/src/llamafactory/extras/logging.py:70
↓ 87 callers
Method
update
(self, val, n=1)
llamafactory/src/llamafactory/extras/misc.py:69
↓ 79 callers
Method
start
(self)
src/data/compressor/dataset_preparation/API/gpt-4o.py:351
↓ 59 callers
Method
warning_rank0
(self, *args, **kwargs)
llamafactory/src/llamafactory/extras/logging.py:73
↓ 46 callers
Function
get_mm_plugin
r"""Get plugin for multimodal inputs.
llamafactory/src/llamafactory/data/mm_plugin.py:1905
↓ 38 callers
Function
check_version
r"""Optionally check the package version.
llamafactory/src/llamafactory/extras/misc.py:76
↓ 35 callers
Method
get_elem_by_id
r"""Get element by id. Example: top.lang, train.dataset
llamafactory/src/llamafactory/webui/manager.py:46
↓ 30 callers
Method
_validate_input
r"""Validate if this model accepts the input modalities.
llamafactory/src/llamafactory/data/mm_plugin.py:147
↓ 29 callers
Method
update
(self, n=1)
src/data/compressor/dataset_preparation/API/api_result_completiness/retry_empty_chunks.py:366
↓ 28 callers
Method
apply
r"""Forms a list of slots according to the inputs to encode.
llamafactory/src/llamafactory/data/formatter.py:33
↓ 27 callers
Function
get_template_and_fix_tokenizer
r"""Get chat template and fixes the tokenizer.
llamafactory/src/llamafactory/data/template.py:591
↓ 26 callers
Function
load_tokenizer
r"""Load pretrained tokenizer and optionally loads processor. Note: including inplace operation of model_args.
llamafactory/src/llamafactory/model/loader.py:71
↓ 23 callers
Method
close
(self)
src/data/compressor/dataset_preparation/API/api_result_completiness/retry_empty_chunks.py:368
↓ 22 callers
Function
_register_composite_model
r"""Register a new composite model. Args: model_type: model type projector_key: multi_modal_projector vision_model_keys:
llamafactory/src/llamafactory/model/model_utils/visual.py:58
↓ 20 callers
Function
is_env_enabled
r"""Check if the environment variable is enabled.
llamafactory/src/llamafactory/extras/misc.py:231
↓ 18 callers
Method
_get_mm_inputs
( self, images: list["ImageInput"], videos: list["VideoInput"], audios: list["
llamafactory/src/llamafactory/data/mm_plugin.py:1534
↓ 17 callers
Method
_validate_messages
r"""Validate if the number of images, videos and audios match the number of placeholders in messages.
llamafactory/src/llamafactory/data/mm_plugin.py:187
↓ 17 callers
Function
get_save_dir
r"""Get the path to saved model checkpoints.
llamafactory/src/llamafactory/webui/common.py:59
↓ 17 callers
Function
tlog
(msg: str)
src/data/compressor/dataset_preparation/API/gpt4o_ranges_labeler.py:188
↓ 16 callers
Function
_is_package_available
(name: str)
llamafactory/src/llamafactory/extras/packages.py:30
↓ 16 callers
Function
load_model
r"""Load pretrained model.
llamafactory/src/llamafactory/model/loader.py:131
↓ 16 callers
Function
tlog
(msg: str)
src/data/compressor/dataset_preparation/API/gpt-4o.py:189
↓ 14 callers
Function
_set_z3_leaf_modules
(model: "PreTrainedModel", leaf_modules: list[Union["nn.Module", str]])
llamafactory/src/llamafactory/model/model_utils/moe.py:29
↓ 14 callers
Method
eval
(self)
llamafactory/src/llamafactory/eval/evaluator.py:79
↓ 14 callers
Function
is_transformers_version_greater_than
(content: str)
llamafactory/src/llamafactory/extras/packages.py:94
↓ 14 callers
Function
load_train_model
(add_valuehead: bool = False, **kwargs)
llamafactory/src/llamafactory/train/test_utils.py:64
↓ 14 callers
Method
process_messages
r"""Pre-process input messages before tokenization for VLMs.
llamafactory/src/llamafactory/data/mm_plugin.py:395
↓ 14 callers
Method
to_dict
(self)
llamafactory/src/llamafactory/hparams/data_args.py:185
↓ 13 callers
Method
encode_oneturn
r"""Return a single pair of token ids representing prompt and response respectively.
llamafactory/src/llamafactory/data/template.py:58
↓ 12 callers
Function
_check_plugin
( plugin: "BasePlugin", tokenizer: "PreTrainedTokenizer", processor: "ProcessorMixin", expecte
llamafactory/tests/data/test_mm_plugin.py:128
↓ 12 callers
Function
_load_tokenizer_module
(model_name_or_path: str)
llamafactory/tests/data/test_mm_plugin.py:123
↓ 12 callers
Function
get_dataset
r"""Get the train dataset and optionally gets the evaluation dataset.
llamafactory/src/llamafactory/data/loader.py:277
↓ 12 callers
Function
is_gradio_available
()
llamafactory/src/llamafactory/extras/packages.py:65
↓ 12 callers
Function
to_device
(x, device)
src/data/compressor_longformer/train/train_longformer_v2.py:57
↓ 11 callers
Method
process_token_ids
r"""Pre-process token ids after tokenization for VLMs.
llamafactory/src/llamafactory/data/mm_plugin.py:407
↓ 11 callers
Function
tlog
(msg: str)
src/data/compressor/dataset_preparation/API/api_result_completiness/retry_empty_chunks.py:150
↓ 10 callers
Function
_get_mm_inputs
(processor: "ProcessorMixin")
llamafactory/tests/data/test_mm_plugin.py:86
↓ 10 callers
Method
_regularize_images
r"""Regularize images to avoid error. Including reading and pre-processing.
llamafactory/src/llamafactory/data/mm_plugin.py:247
↓ 10 callers
Function
is_mathy_token
(t: Token)
src/data/compressor/dataset_preparation/camel/camel_chunk_cot_and_answer/math_tokenizer_no_space.py:174
↓ 10 callers
Function
load_dataset_module
(**kwargs)
llamafactory/src/llamafactory/train/test_utils.py:105
↓ 10 callers
Method
step
(self, closure: Optional[Callable[[], float]] = None)
llamafactory/src/llamafactory/train/trainer_utils.py:81
↓ 9 callers
Method
_find_medias
r"""Optionally concatenate media path to media dir when loading from local disk.
llamafactory/src/llamafactory/data/converter.py:43
↓ 9 callers
Method
add_elems
r"""Add elements to manager.
llamafactory/src/llamafactory/webui/manager.py:30
↓ 9 callers
Method
apply
(self, **kwargs)
llamafactory/src/llamafactory/data/formatter.py:98
↓ 9 callers
Method
extract
(self, content: str)
llamafactory/src/llamafactory/data/formatter.py:144
↓ 9 callers
Function
is_ddp
()
src/data/compressor_longformer/train/train_longformer_v2.py:43
↓ 9 callers
Function
shift
(state: "torch.Tensor")
llamafactory/src/llamafactory/model/model_utils/longlora.py:96
↓ 9 callers
Method
warning_rank0_once
(self, *args, **kwargs)
llamafactory/src/llamafactory/extras/logging.py:76
↓ 8 callers
Method
_convert_slots_to_jinja
r"""Convert slots to jinja template.
llamafactory/src/llamafactory/data/template.py:215
↓ 8 callers
Method
get_token_count
获取文本的token数量
src/data/compressor/dataset_preparation/camel/camel_chunk_cot_and_answer/claude_chunk_camel_data.py:30
↓ 8 callers
Function
get_train_args
(args: Optional[Union[dict[str, Any], list[str]]] = None)
llamafactory/src/llamafactory/hparams/parser.py:209
↓ 8 callers
Function
load_reference_model
( model_path: str, lora_path: Optional[str] = None, use_lora: bool = False, use_pissa: bool =
llamafactory/src/llamafactory/train/test_utils.py:76
↓ 8 callers
Method
set_attr
(self, key: str, obj: dict[str, Any], default: Optional[Any] = None)
llamafactory/src/llamafactory/data/parser.py:69
↓ 8 callers
Function
to_device
(x, device)
src/data/compressor_longformer/train/train_longformer.py:81
↓ 7 callers
Function
_check_template
r"""Check template. Args: model_id: the model id on hugging face hub. template_name: the template name. prompt_str: the s
llamafactory/tests/data/test_template.py:63
↓ 7 callers
Method
_regularize_videos
r"""Regularizes videos to avoid error. Including reading, resizing and converting.
llamafactory/src/llamafactory/data/mm_plugin.py:268
↓ 7 callers
Function
compare_model
(model_a: "torch.nn.Module", model_b: "torch.nn.Module", diff_keys: list[str] = [])
llamafactory/src/llamafactory/train/test_utils.py:35
↓ 7 callers
Function
get_device_count
r"""Get the number of available devices.
llamafactory/src/llamafactory/extras/misc.py:160
↓ 7 callers
Function
get_infer_args
(args: Optional[Union[dict[str, Any], list[str]]] = None)
llamafactory/src/llamafactory/hparams/parser.py:425
↓ 7 callers
Method
get_token_count
获取文本的token数量
src/data/compressor/dataset_preparation/camel/camel_chunk_cot_and_answer/chunk_camel_data.py:32
↓ 7 callers
Function
is_main
()
src/data/compressor_longformer/train/train_longformer_v2.py:46
↓ 7 callers
Function
load_config
r"""Load user config if exists.
llamafactory/src/llamafactory/webui/common.py:74
↓ 7 callers
Function
load_infer_model
(add_valuehead: bool = False, **kwargs)
llamafactory/src/llamafactory/train/test_utils.py:70
↓ 7 callers
Method
save_model
r"""Save model checkpoint. Subclass and override to inject custom behavior.
llamafactory/src/llamafactory/train/ppo/trainer.py:477
↓ 7 callers
Function
torch_gc
r"""Collect the device memory.
llamafactory/src/llamafactory/extras/misc.py:254
↓ 6 callers
Function
_norm
(s: str)
src/eval/qwen1p7b_eval/recheck_accuracy.py:198
↓ 6 callers
Function
_norm
(s: str)
src/eval/qwen1p7b_eval/mmlu-stem/recheck_acc.py:182
↓ 6 callers
Function
available_ratios_for_id
(idx: Dict[float, Dict[str, dict]], sid: str)
src/training/qwen1p7b_rl/dataset/merge_data.py:122
↓ 6 callers
Function
create_custom_optimizer
( model: "PreTrainedModel", training_args: "TrainingArguments", finetuning_args: "FinetuningArgume
llamafactory/src/llamafactory/train/trainer_utils.py:522
↓ 6 callers
Function
create_custom_scheduler
( training_args: "TrainingArguments", num_training_steps: int, optimizer: Optional["torch.optim.Op
llamafactory/src/llamafactory/train/trainer_utils.py:546
↓ 6 callers
Function
emit_sample
(r_target: float)
src/data/compressor_longformer/validate/build_mixture_llamafactory.py:294
↓ 6 callers
Function
get_current_device
r"""Get the current available device.
llamafactory/src/llamafactory/extras/misc.py:144
↓ 6 callers
Method
get_stop_token_ids
r"""Return stop token ids.
llamafactory/src/llamafactory/data/template.py:89
↓ 6 callers
Function
infer_seqlen
r"""Compute the real sequence length after truncation by the cutoff_len.
llamafactory/src/llamafactory/data/processor/processor_utils.py:76
↓ 6 callers
Function
is_main_process
()
src/data/compressor_longformer/train/train_longformer.py:62
↓ 6 callers
Function
load_json
(file, encoding='utf-8')
src/data/compressor_longformer/validate/Tokenskip_pipeline/build_llamafactory_input_numeric.py:6
↓ 6 callers
Function
numpify
r"""Cast a torch tensor or a numpy array to a numpy array.
llamafactory/src/llamafactory/extras/misc.py:236
↓ 6 callers
Function
pick_by_probs
(items: List[float], probs: Dict[float, float] = None)
src/data/compressor_longformer/validate/build_mixture_llamafactory.py:230
↓ 6 callers
Function
plot_loss
r"""Plot loss curves and saves the image.
llamafactory/src/llamafactory/extras/ploting.py:69
↓ 6 callers
Function
save_result_to_file
(result: Dict[str, Any], output_file: str, lock_file: str)
src/training/qwen1p7b_rl/dataset/vllm_inference.py:285
↓ 6 callers
Function
use_modelscope
()
llamafactory/src/llamafactory/extras/misc.py:304
↓ 6 callers
Function
use_openmind
()
llamafactory/src/llamafactory/extras/misc.py:308
↓ 5 callers
Method
__post_init__
(self)
llamafactory/src/llamafactory/hparams/model_args.py:342
↓ 5 callers
Function
_as_ratio
Accept '20' or '0.2' → 0.2
src/training/qwen1p7b_rl/dataset/rl_model_ratio_v2/convert_data_to_verl.py:101
↓ 5 callers
Function
_check_tokenization
r"""Check token ids and texts. encode(text) == token_ids decode(token_ids) == text
llamafactory/tests/data/test_template.py:50
↓ 5 callers
Method
_get_mm_inputs
( self, images: list["ImageInput"], videos: list["VideoInput"], audios: list["
llamafactory/src/llamafactory/data/mm_plugin.py:925
↓ 5 callers
Method
_regularize_audios
r"""Regularizes audios to avoid error. Including reading and resampling.
llamafactory/src/llamafactory/data/mm_plugin.py:292
↓ 5 callers
Function
add_span
(a,b)
src/data/compressor/dataset_preparation/camel/camel_chunk_cot_and_answer/math_tokenizer_no_space.py:464
↓ 5 callers
Method
apply
(self, **kwargs)
llamafactory/src/llamafactory/data/formatter.py:135
↓ 5 callers
Function
counts_to_metrics
(c: Dict[str, int])
src/data/compressor_longformer/train/train_longformer_v2.py:198
↓ 5 callers
Function
create_modelcard_and_push
( trainer: "Trainer", model_args: "ModelArguments", data_args: "DataArguments", training_args:
llamafactory/src/llamafactory/train/trainer_utils.py:85
↓ 5 callers
Function
expand_minimal
Expand around anchor to include a minimal mathy expression. Stops at sentence separators and at **top-level commas** between tokens.
src/data/compressor/dataset_preparation/camel/camel_chunk_cot_and_answer/math_tokenizer_no_space.py:246
↓ 5 callers
Method
get_mm_inputs
r"""Build batched multimodal inputs for VLMs. Arguments: images: a list of image inputs, shape (num_images,) videos:
llamafactory/src/llamafactory/data/mm_plugin.py:421
↓ 5 callers
Method
get_token_count
(self, text: str)
src/data/compressor/dataset_preparation/camel/camel_chunk_cot_and_answer/gpt_chunk_camel_data.py:410
↓ 5 callers
Method
get_token_count
(self, text: str)
src/data/compressor/dataset_preparation/camel/camel_chunk_only_cot/gpt_chunk_camel_data.py:410
↓ 5 callers
Function
is_nonempty_ranges
(r)
src/data/compressor/dataset_preparation/API/api_result_completiness/merge_results.py:20
next →
1–100 of 1,850, ranked by callers