跳转至

mint.types

mint.types 模块中的所有公开类型,按功能分组。

数据与输入

用于准备训练数据和模型输入的类型。

类型 说明
Datum 单条训练样本,包含输入 token 和 loss mask
ModelInput 用于采样或训练的 token 序列输入
TensorData 带 dtype 和 shape 元数据的张量容器
EncodedTextChunk 编码后的文本 token 块
DmelChunk 客户端编码的 DMel 音频输入,适用于支持音频的模型
ImageChunk 用于视觉语言模型的图像数据块
ImageAssetPointerChunk 指向已上传图像资产的指针

训练

用于配置和观测训练操作的类型。

类型 说明
ForwardBackwardOutput 前向-反向传播的结果(loss、梯度、指标)
OptimStepResponse 优化器步骤的结果
AdamParams Adam 优化器超参数(learning rate、betas、epsilon)
LoraConfig LoRA adapter 配置(秩、训练哪些层)
GetInfoResponse 训练客户端模型的信息
ModelData 模型架构与配置的元数据

采样与推理

用于文本生成和采样结果的类型。

类型 说明
SamplingParams 采样配置(temperature、max tokens、stop sequences)
SampleResponse 采样请求的响应
SampledSequence 单条生成序列,包含 token 和对数概率
TopkPromptLogprobs prompt token 的 top-k 对数概率

模型与 checkpoint

用于管理模型、checkpoint 和训练任务的类型。

类型 说明
Checkpoint 已保存的模型 checkpoint
ParsedCheckpointTinkerPath mindlab-toolkit:// checkpoint 路径解析后的各部分
TrainingRun 训练任务的元数据
SupportedModel 服务端支持的模型信息
GetServerCapabilitiesResponse 服务端能力,包括支持的模型
CheckpointsListResponse 分页的 checkpoint 列表
CheckpointArchiveUrlResponse 下载 checkpoint 归档的 URL
TrainingRunsResponse 分页的训练任务列表
Cursor 列表响应的分页游标
SaveWeightsResponse 保存模型权重的响应
SaveWeightsForSamplerResponse 为采样器保存权重的响应
WeightsInfoResponse 加载公开 checkpoint 所需的最少信息

认证与客户端配置

用于认证和客户端配置的类型。

类型 说明
AuthTokenResponse 认证令牌请求的响应
ClientConfigRequest 客户端配置请求
ClientConfigResponse 服务端提供的客户端配置