MCPcopy Create free account

hub / github.com/declare-lab/resta / types & classes

Types & classes136 in github.com/declare-lab/resta

↓ 57 callersClassStringFormatter
sft/src/llmtuner/data/formatter.py:91
↓ 17 callersClassEmptyFormatter
sft/src/llmtuner/data/formatter.py:85
↓ 12 callersClassExecutor
Schedules and executes a set of tasks and their dependencies. Handles scheduling, execution, the movement of data between devices, and the l
merge/mergekit/graph.py:110
↓ 11 callersClassConfigParameterDef
merge/mergekit/merge_methods/base.py:27
↓ 7 callersClassMergeConfiguration
merge/mergekit/config.py:84
↓ 6 callersClassConfigReader
merge/mergekit/config.py:119
↓ 6 callersClassInputModelDefinition
merge/mergekit/config.py:72
↓ 6 callersClassInputSliceDefinition
merge/mergekit/config.py:66
↓ 5 callersClassChatCompletionMessage
sft/src/llmtuner/api/protocol.py:53
↓ 5 callersClassImmutableMap
merge/mergekit/common.py:234
↓ 5 callersClassTensorWriter
merge/mergekit/io/tensor_writer.py:25
↓ 4 callersClassChatModel
sft/src/llmtuner/chat/chat_model.py:22
↓ 4 callersClassDatasetAttr
r""" Dataset attributes.
sft/src/llmtuner/data/parser.py:15
↓ 4 callersClassGeneralizedTaskArithmeticMerge
merge/mergekit/merge_methods/generalized_task_arithmetic.py:37
↓ 4 callersClassLazyTensorLoader
merge/mergekit/io/lazy_tensor_loader.py:114
↓ 4 callersClassMergeOptions
merge/mergekit/options.py:26
↓ 4 callersClassOutputSliceDefinition
merge/mergekit/config.py:77
↓ 3 callersClassChatCompletionResponseStreamChoice
sft/src/llmtuner/api/protocol.py:77
↓ 3 callersClassChatCompletionStreamResponse
sft/src/llmtuner/api/protocol.py:98
↓ 3 callersClassLoaderCache
merge/mergekit/io/tasks.py:14
↓ 2 callersClassAverageMeter
r""" Computes and stores the average and current value.
sft/src/llmtuner/extras/misc.py:38
↓ 2 callersClassEngine
sft/src/llmtuner/webui/engine.py:14
↓ 2 callersClassEvaluator
sft/src/llmtuner/eval/evaluator.py:21
↓ 2 callersClassFinetuningArguments
r""" Arguments pertaining to which techniques we are going to fine-tuning with.
sft/src/llmtuner/hparams/finetuning_args.py:160
↓ 2 callersClassFixValueHeadModelCallback
sft/src/llmtuner/extras/callbacks.py:22
↓ 2 callersClassFunctionFormatter
sft/src/llmtuner/data/formatter.py:111
↓ 2 callersClassLogCallback
sft/src/llmtuner/extras/callbacks.py:35
↓ 2 callersClassModelArguments
r""" Arguments pertaining to which model/config/tokenizer we are going to fine-tune.
sft/src/llmtuner/hparams/model_args.py:6
↓ 2 callersClassShardInfo
merge/mergekit/io/lazy_tensor_loader.py:32
↓ 1 callersClassBuildTokenizer
merge/mergekit/tokenizer.py:266
↓ 1 callersClassCMMLUConfig
sft/evaluation/cmmlu/cmmlu.py:110
↓ 1 callersClassCevalConfig
sft/evaluation/ceval/ceval.py:95
↓ 1 callersClassChatCompletionResponse
sft/src/llmtuner/api/protocol.py:89
↓ 1 callersClassChatCompletionResponseChoice
sft/src/llmtuner/api/protocol.py:71
↓ 1 callersClassChatCompletionResponseUsage
sft/src/llmtuner/api/protocol.py:83
↓ 1 callersClassCircularTask
merge/tests/test_graph.py:193
↓ 1 callersClassComputeMetrics
r""" Wraps the tokenizer into metric functions, used in Seq2SeqPeftTrainer.
sft/src/llmtuner/train/sft/metric.py:24
↓ 1 callersClassConditionalParameter
merge/mergekit/config.py:27
↓ 1 callersClassConfiguredArchitectureInfo
merge/mergekit/architecture.py:123
↓ 1 callersClassCustomDPOTrainer
sft/src/llmtuner/train/dpo/trainer.py:17
↓ 1 callersClassCustomPPOTrainer
r""" Inherits PPOTrainer.
sft/src/llmtuner/train/ppo/trainer.py:31
↓ 1 callersClassCustomSeq2SeqTrainer
r""" Inherits PeftTrainer to compute generative metrics such as BLEU and ROUGE.
sft/src/llmtuner/train/sft/trainer.py:21
↓ 1 callersClassDPODataCollatorWithPadding
r""" Data collator for pairwise data.
sft/src/llmtuner/train/dpo/collator.py:9
↓ 1 callersClassDeferredLoad
merge/mergekit/io/lazy_unpickle.py:55
↓ 1 callersClassDumbPytorchLoader
Naive `torch.load` shard loading.
merge/mergekit/io/loader.py:76
↓ 1 callersClassDummyTask
merge/tests/test_graph.py:12
↓ 1 callersClassEvalTemplate
sft/src/llmtuner/eval/template.py:13
↓ 1 callersClassFinalizeModel
merge/mergekit/io/tasks.py:163
↓ 1 callersClassFunction
sft/src/llmtuner/api/protocol.py:37
↓ 1 callersClassFunctionCall
sft/src/llmtuner/api/protocol.py:42
↓ 1 callersClassGTATask
merge/mergekit/merge_methods/generalized_task_arithmetic.py:75
↓ 1 callersClassGatherTensors
merge/mergekit/io/tasks.py:95
↓ 1 callersClassGeneratingArguments
r""" Arguments pertaining to specify the decoding parameters.
sft/src/llmtuner/hparams/generating_args.py:6
↓ 1 callersClassJsonArchitectureInfo
merge/mergekit/architecture.py:165
↓ 1 callersClassLazyPickleLoader
Loader for pytorch files using a custom unpickler and vigorous monkeypatching.
merge/mergekit/io/loader.py:53
↓ 1 callersClassLinearMerge
merge/mergekit/merge_methods/linear.py:66
↓ 1 callersClassLinearMergeTask
merge/mergekit/merge_methods/linear.py:28
↓ 1 callersClassLoadTensor
merge/mergekit/io/tasks.py:50
↓ 1 callersClassLoggerHandler
r""" Logger handler used in Web UI.
sft/src/llmtuner/extras/logging.py:5
↓ 1 callersClassMMLUConfig
sft/evaluation/mmlu/mmlu.py:100
↓ 1 callersClassManager
sft/src/llmtuner/webui/manager.py:8
↓ 1 callersClassMergePlanner
merge/mergekit/plan.py:41
↓ 1 callersClassMixtralTensorNames
merge/mergekit/architecture.py:246
↓ 1 callersClassModelCard
sft/src/llmtuner/api/protocol.py:25
↓ 1 callersClassModelList
sft/src/llmtuner/api/protocol.py:32
↓ 1 callersClassModelReference
A reference to a language model. Can be a hf hub path (username/repo), or local. Optionally includes a LoRA.
merge/mergekit/common.py:77
↓ 1 callersClassPairwiseDataCollatorWithPadding
r""" Data collator for pairwise data.
sft/src/llmtuner/train/rm/collator.py:9
↓ 1 callersClassPairwiseTrainer
r""" Inherits PeftTrainer to compute pairwise loss.
sft/src/llmtuner/train/rm/trainer.py:19
↓ 1 callersClassPassthroughMerge
merge/mergekit/merge_methods/passthrough.py:45
↓ 1 callersClassPassthroughMergeTask
merge/mergekit/merge_methods/passthrough.py:27
↓ 1 callersClassResponse
sft/src/llmtuner/chat/chat_model.py:15
↓ 1 callersClassRunner
sft/src/llmtuner/webui/runner.py:26
↓ 1 callersClassSaveTensor
merge/mergekit/io/tasks.py:144
↓ 1 callersClassScoreEvaluationResponse
sft/src/llmtuner/api/protocol.py:112
↓ 1 callersClassShardSizeParamType
merge/mergekit/options.py:59
↓ 1 callersClassShardedTensorIndex
merge/mergekit/io/lazy_tensor_loader.py:38
↓ 1 callersClassShell
sft/tests/loftq_init.py:20
↓ 1 callersClassSlerpMerge
merge/mergekit/merge_methods/slerp.py:67
↓ 1 callersClassSlerpTask
merge/mergekit/merge_methods/slerp.py:29
↓ 1 callersClassTemplateWithArithmetic
merge/mergekit/architecture.py:161
↓ 1 callersClassTensorWriterTask
merge/mergekit/io/tasks.py:128
↓ 1 callersClassTokenizerInfo
merge/mergekit/tokenizer.py:261
↓ 1 callersClassTokenizerPermutationMerge
merge/mergekit/merge_methods/tokenizer_permute.py:121
↓ 1 callersClassTokenizerPermutationMergeTask
merge/mergekit/merge_methods/tokenizer_permute.py:30
↓ 1 callersClassToolFormatter
sft/src/llmtuner/data/formatter.py:135
↓ 1 callersClassTorchArchiveReader
Class for lazily reading (sections of) files from a torch ZIP archive. Maintains a handle to the most recently opened file for faster access
merge/mergekit/io/lazy_unpickle.py:118
↓ 1 callersClassWebChatModel
sft/src/llmtuner/webui/chatter.py:19
↓ 1 callersClassWeightInfo
Information about an individual weight tensor in a model. Attributes: name (str): The name of the tensor representing the wei
merge/mergekit/architecture.py:28
ClassArchitectureInfo
merge/mergekit/architecture.py:69
ClassBakllamaConfig
merge/mergekit/scripts/bakllama.py:38
ClassBelleMultiturn
sft/data/belle_multiturn/belle_multiturn.py:21
ClassCMMLU
sft/evaluation/cmmlu/cmmlu.py:115
ClassCeval
sft/evaluation/ceval/ceval.py:100
ClassChatCompletionRequest
sft/src/llmtuner/api/protocol.py:59
ClassChatMessage
sft/src/llmtuner/api/protocol.py:48
ClassConfigYamlDumper
Custom YAML dumper to format lists of numbers in flow style.
merge/mergekit/config.py:199
ClassConsensusMethod
merge/mergekit/merge_methods/generalized_task_arithmetic.py:32
ClassDataArguments
r""" Arguments pertaining to what data we are going to input our model for training and evaluation.
sft/src/llmtuner/hparams/data_args.py:6
ClassDownloadSource
sft/src/llmtuner/extras/constants.py:50
ClassEvaluationArguments
r""" Arguments pertaining to specify the evaluation parameters.
sft/src/llmtuner/hparams/evaluation_args.py:9
next →1–100 of 136, ranked by callers