Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/declare-lab/resta
/ types & classes
Types & classes
136 in github.com/declare-lab/resta
⨍
Functions
581
◇
Types & classes
136
↳
Endpoints
5
↓ 57 callers
Class
StringFormatter
sft/src/llmtuner/data/formatter.py:91
↓ 17 callers
Class
EmptyFormatter
sft/src/llmtuner/data/formatter.py:85
↓ 12 callers
Class
Executor
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 callers
Class
ConfigParameterDef
merge/mergekit/merge_methods/base.py:27
↓ 7 callers
Class
MergeConfiguration
merge/mergekit/config.py:84
↓ 6 callers
Class
ConfigReader
merge/mergekit/config.py:119
↓ 6 callers
Class
InputModelDefinition
merge/mergekit/config.py:72
↓ 6 callers
Class
InputSliceDefinition
merge/mergekit/config.py:66
↓ 5 callers
Class
ChatCompletionMessage
sft/src/llmtuner/api/protocol.py:53
↓ 5 callers
Class
ImmutableMap
merge/mergekit/common.py:234
↓ 5 callers
Class
TensorWriter
merge/mergekit/io/tensor_writer.py:25
↓ 4 callers
Class
ChatModel
sft/src/llmtuner/chat/chat_model.py:22
↓ 4 callers
Class
DatasetAttr
r""" Dataset attributes.
sft/src/llmtuner/data/parser.py:15
↓ 4 callers
Class
GeneralizedTaskArithmeticMerge
merge/mergekit/merge_methods/generalized_task_arithmetic.py:37
↓ 4 callers
Class
LazyTensorLoader
merge/mergekit/io/lazy_tensor_loader.py:114
↓ 4 callers
Class
MergeOptions
merge/mergekit/options.py:26
↓ 4 callers
Class
OutputSliceDefinition
merge/mergekit/config.py:77
↓ 3 callers
Class
ChatCompletionResponseStreamChoice
sft/src/llmtuner/api/protocol.py:77
↓ 3 callers
Class
ChatCompletionStreamResponse
sft/src/llmtuner/api/protocol.py:98
↓ 3 callers
Class
LoaderCache
merge/mergekit/io/tasks.py:14
↓ 2 callers
Class
AverageMeter
r""" Computes and stores the average and current value.
sft/src/llmtuner/extras/misc.py:38
↓ 2 callers
Class
Engine
sft/src/llmtuner/webui/engine.py:14
↓ 2 callers
Class
Evaluator
sft/src/llmtuner/eval/evaluator.py:21
↓ 2 callers
Class
FinetuningArguments
r""" Arguments pertaining to which techniques we are going to fine-tuning with.
sft/src/llmtuner/hparams/finetuning_args.py:160
↓ 2 callers
Class
FixValueHeadModelCallback
sft/src/llmtuner/extras/callbacks.py:22
↓ 2 callers
Class
FunctionFormatter
sft/src/llmtuner/data/formatter.py:111
↓ 2 callers
Class
LogCallback
sft/src/llmtuner/extras/callbacks.py:35
↓ 2 callers
Class
ModelArguments
r""" Arguments pertaining to which model/config/tokenizer we are going to fine-tune.
sft/src/llmtuner/hparams/model_args.py:6
↓ 2 callers
Class
ShardInfo
merge/mergekit/io/lazy_tensor_loader.py:32
↓ 1 callers
Class
BuildTokenizer
merge/mergekit/tokenizer.py:266
↓ 1 callers
Class
CMMLUConfig
sft/evaluation/cmmlu/cmmlu.py:110
↓ 1 callers
Class
CevalConfig
sft/evaluation/ceval/ceval.py:95
↓ 1 callers
Class
ChatCompletionResponse
sft/src/llmtuner/api/protocol.py:89
↓ 1 callers
Class
ChatCompletionResponseChoice
sft/src/llmtuner/api/protocol.py:71
↓ 1 callers
Class
ChatCompletionResponseUsage
sft/src/llmtuner/api/protocol.py:83
↓ 1 callers
Class
CircularTask
merge/tests/test_graph.py:193
↓ 1 callers
Class
ComputeMetrics
r""" Wraps the tokenizer into metric functions, used in Seq2SeqPeftTrainer.
sft/src/llmtuner/train/sft/metric.py:24
↓ 1 callers
Class
ConditionalParameter
merge/mergekit/config.py:27
↓ 1 callers
Class
ConfiguredArchitectureInfo
merge/mergekit/architecture.py:123
↓ 1 callers
Class
CustomDPOTrainer
sft/src/llmtuner/train/dpo/trainer.py:17
↓ 1 callers
Class
CustomPPOTrainer
r""" Inherits PPOTrainer.
sft/src/llmtuner/train/ppo/trainer.py:31
↓ 1 callers
Class
CustomSeq2SeqTrainer
r""" Inherits PeftTrainer to compute generative metrics such as BLEU and ROUGE.
sft/src/llmtuner/train/sft/trainer.py:21
↓ 1 callers
Class
DPODataCollatorWithPadding
r""" Data collator for pairwise data.
sft/src/llmtuner/train/dpo/collator.py:9
↓ 1 callers
Class
DeferredLoad
merge/mergekit/io/lazy_unpickle.py:55
↓ 1 callers
Class
DumbPytorchLoader
Naive `torch.load` shard loading.
merge/mergekit/io/loader.py:76
↓ 1 callers
Class
DummyTask
merge/tests/test_graph.py:12
↓ 1 callers
Class
EvalTemplate
sft/src/llmtuner/eval/template.py:13
↓ 1 callers
Class
FinalizeModel
merge/mergekit/io/tasks.py:163
↓ 1 callers
Class
Function
sft/src/llmtuner/api/protocol.py:37
↓ 1 callers
Class
FunctionCall
sft/src/llmtuner/api/protocol.py:42
↓ 1 callers
Class
GTATask
merge/mergekit/merge_methods/generalized_task_arithmetic.py:75
↓ 1 callers
Class
GatherTensors
merge/mergekit/io/tasks.py:95
↓ 1 callers
Class
GeneratingArguments
r""" Arguments pertaining to specify the decoding parameters.
sft/src/llmtuner/hparams/generating_args.py:6
↓ 1 callers
Class
JsonArchitectureInfo
merge/mergekit/architecture.py:165
↓ 1 callers
Class
LazyPickleLoader
Loader for pytorch files using a custom unpickler and vigorous monkeypatching.
merge/mergekit/io/loader.py:53
↓ 1 callers
Class
LinearMerge
merge/mergekit/merge_methods/linear.py:66
↓ 1 callers
Class
LinearMergeTask
merge/mergekit/merge_methods/linear.py:28
↓ 1 callers
Class
LoadTensor
merge/mergekit/io/tasks.py:50
↓ 1 callers
Class
LoggerHandler
r""" Logger handler used in Web UI.
sft/src/llmtuner/extras/logging.py:5
↓ 1 callers
Class
MMLUConfig
sft/evaluation/mmlu/mmlu.py:100
↓ 1 callers
Class
Manager
sft/src/llmtuner/webui/manager.py:8
↓ 1 callers
Class
MergePlanner
merge/mergekit/plan.py:41
↓ 1 callers
Class
MixtralTensorNames
merge/mergekit/architecture.py:246
↓ 1 callers
Class
ModelCard
sft/src/llmtuner/api/protocol.py:25
↓ 1 callers
Class
ModelList
sft/src/llmtuner/api/protocol.py:32
↓ 1 callers
Class
ModelReference
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 callers
Class
PairwiseDataCollatorWithPadding
r""" Data collator for pairwise data.
sft/src/llmtuner/train/rm/collator.py:9
↓ 1 callers
Class
PairwiseTrainer
r""" Inherits PeftTrainer to compute pairwise loss.
sft/src/llmtuner/train/rm/trainer.py:19
↓ 1 callers
Class
PassthroughMerge
merge/mergekit/merge_methods/passthrough.py:45
↓ 1 callers
Class
PassthroughMergeTask
merge/mergekit/merge_methods/passthrough.py:27
↓ 1 callers
Class
Response
sft/src/llmtuner/chat/chat_model.py:15
↓ 1 callers
Class
Runner
sft/src/llmtuner/webui/runner.py:26
↓ 1 callers
Class
SaveTensor
merge/mergekit/io/tasks.py:144
↓ 1 callers
Class
ScoreEvaluationResponse
sft/src/llmtuner/api/protocol.py:112
↓ 1 callers
Class
ShardSizeParamType
merge/mergekit/options.py:59
↓ 1 callers
Class
ShardedTensorIndex
merge/mergekit/io/lazy_tensor_loader.py:38
↓ 1 callers
Class
Shell
sft/tests/loftq_init.py:20
↓ 1 callers
Class
SlerpMerge
merge/mergekit/merge_methods/slerp.py:67
↓ 1 callers
Class
SlerpTask
merge/mergekit/merge_methods/slerp.py:29
↓ 1 callers
Class
TemplateWithArithmetic
merge/mergekit/architecture.py:161
↓ 1 callers
Class
TensorWriterTask
merge/mergekit/io/tasks.py:128
↓ 1 callers
Class
TokenizerInfo
merge/mergekit/tokenizer.py:261
↓ 1 callers
Class
TokenizerPermutationMerge
merge/mergekit/merge_methods/tokenizer_permute.py:121
↓ 1 callers
Class
TokenizerPermutationMergeTask
merge/mergekit/merge_methods/tokenizer_permute.py:30
↓ 1 callers
Class
ToolFormatter
sft/src/llmtuner/data/formatter.py:135
↓ 1 callers
Class
TorchArchiveReader
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 callers
Class
WebChatModel
sft/src/llmtuner/webui/chatter.py:19
↓ 1 callers
Class
WeightInfo
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
Class
ArchitectureInfo
merge/mergekit/architecture.py:69
Class
BakllamaConfig
merge/mergekit/scripts/bakllama.py:38
Class
BelleMultiturn
sft/data/belle_multiturn/belle_multiturn.py:21
Class
CMMLU
sft/evaluation/cmmlu/cmmlu.py:115
Class
Ceval
sft/evaluation/ceval/ceval.py:100
Class
ChatCompletionRequest
sft/src/llmtuner/api/protocol.py:59
Class
ChatMessage
sft/src/llmtuner/api/protocol.py:48
Class
ConfigYamlDumper
Custom YAML dumper to format lists of numbers in flow style.
merge/mergekit/config.py:199
Class
ConsensusMethod
merge/mergekit/merge_methods/generalized_task_arithmetic.py:32
Class
DataArguments
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
Class
DownloadSource
sft/src/llmtuner/extras/constants.py:50
Class
EvaluationArguments
r""" Arguments pertaining to specify the evaluation parameters.
sft/src/llmtuner/hparams/evaluation_args.py:9
next →
1–100 of 136, ranked by callers