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 |
服务端提供的客户端配置 |