145 lines
4.0 KiB
TypeScript
145 lines
4.0 KiB
TypeScript
/**
|
||
* 全局常量与映射表
|
||
*/
|
||
|
||
/** 模型类型 */
|
||
export const MODEL_TYPE_MAP: Record<string, string> = {
|
||
LLM: '大语言模型',
|
||
CV: '计算机视觉',
|
||
NLP: '自然语言处理',
|
||
Embedding: '向量模型',
|
||
Other: '其他',
|
||
}
|
||
|
||
/** 模型用途 */
|
||
export const PURPOSE_MAP: Record<string, { text: string; type: string }> = {
|
||
training: { text: '训练', type: 'primary' },
|
||
inference: { text: '推理', type: 'success' },
|
||
evaluation: { text: '评测', type: 'warning' },
|
||
}
|
||
|
||
/** 模型来源 */
|
||
export const MODEL_SOURCE_MAP: Record<string, string> = {
|
||
local: '本地模型',
|
||
api: '在线模型',
|
||
online: '在线模型',
|
||
}
|
||
|
||
/** 训练方式 */
|
||
export const TRAIN_TYPE_MAP: Record<string, string> = {
|
||
SFT: 'SFT',
|
||
DPO: 'DPO',
|
||
CPT: 'CPT',
|
||
}
|
||
|
||
/** 训练方法 */
|
||
export const TRAIN_METHOD_MAP: Record<string, string> = {
|
||
lora: 'LoRA',
|
||
qlora: 'QLoRA',
|
||
full: '全量微调',
|
||
prefix: 'Prefix Tuning',
|
||
adapter: 'Adapter',
|
||
peft: 'PEFT',
|
||
adalora: 'AdaLoRA',
|
||
longlora: 'LongLoRA',
|
||
dpo: 'DPO',
|
||
cpt: 'CPT',
|
||
}
|
||
|
||
/** 数据集类型 */
|
||
export const DATASET_TYPE_MAP: Record<string, string> = {
|
||
train: '训练数据',
|
||
test: '测试数据',
|
||
eval: '评测数据',
|
||
val: '验证数据',
|
||
other: '其他',
|
||
}
|
||
|
||
/** 存储位置 */
|
||
export const STORAGE_MAP: Record<string, string> = {
|
||
local: '本地存储',
|
||
minio: 'MinIO',
|
||
cloud: '云存储',
|
||
}
|
||
|
||
/** 评测维度指标类型 */
|
||
export const DIMENSION_TYPE_MAP: Record<string, string> = {
|
||
classification: '大模型-分类型',
|
||
metric: '大模型-指标型',
|
||
text_similarity: '规则-文本相似度',
|
||
}
|
||
|
||
/** 学习率调度器类型 */
|
||
export const LR_SCHEDULER_OPTIONS = [
|
||
{ label: 'cosine', value: 'cosine' },
|
||
{ label: 'linear', value: 'linear' },
|
||
{ label: 'constant', value: 'constant' },
|
||
]
|
||
|
||
/** 训练模板分组（按模型系列） */
|
||
export const TEMPLATE_GROUPS: { label: string; options: { label: string; value: string }[] }[] = [
|
||
{
|
||
label: 'Qwen 系列',
|
||
options: [
|
||
{ label: 'qwen', value: 'qwen' },
|
||
{ label: 'qwen2', value: 'qwen2' },
|
||
{ label: 'qwen2_5', value: 'qwen2_5' },
|
||
],
|
||
},
|
||
{
|
||
label: 'Llama 系列',
|
||
options: [
|
||
{ label: 'llama2', value: 'llama2' },
|
||
{ label: 'llama3', value: 'llama3' },
|
||
{ label: 'llama3_1', value: 'llama3_1' },
|
||
],
|
||
},
|
||
{
|
||
label: 'DeepSeek 系列',
|
||
options: [
|
||
{ label: 'deepseek', value: 'deepseek' },
|
||
{ label: 'deepseek2', value: 'deepseek2' },
|
||
],
|
||
},
|
||
{
|
||
label: '其他系列',
|
||
options: [
|
||
{ label: 'glm4', value: 'glm4' },
|
||
{ label: 'chatglm3', value: 'chatglm3' },
|
||
{ label: 'gemma', value: 'gemma' },
|
||
{ label: 'gemma2', value: 'gemma2' },
|
||
{ label: 'phi', value: 'phi' },
|
||
{ label: 'intern2', value: 'intern2' },
|
||
{ label: 'mistral', value: 'mistral' },
|
||
{ label: 'default', value: 'default' },
|
||
],
|
||
},
|
||
]
|
||
|
||
/** 默认工具列表（其他工具页） */
|
||
export const DEFAULT_TOOLS = [
|
||
{ id: 'data-generate', name: '数据生成', icon: 'fa-database', description: '基于LLM生成微调数据集' },
|
||
{ id: 'json2jsonl', name: 'JSON转JSONL', icon: 'fa-code', description: '将JSON文件转换为JSONL格式' },
|
||
{ id: 'md-convert', name: '转换Markdown', icon: 'fa-file-text', description: '将Markdown文件转换为训练数据' },
|
||
]
|
||
|
||
/** 自定义工具可选图标 */
|
||
export const TOOL_ICONS = [
|
||
'fa-cog', 'fa-database', 'fa-code', 'fa-file-text', 'fa-wrench', 'fa-cube',
|
||
'fa-line-chart', 'fa-bar-chart', 'fa-server', 'fa-cogs', 'fa-tasks', 'fa-trophy',
|
||
'fa-sliders', 'fa-rocket', 'fa-cloud', 'fa-cloud-download', 'fa-microchip',
|
||
'fa-hdd-o', 'fa-globe', 'fa-info-circle', 'fa-question-circle-o', 'fa-user-o',
|
||
'fa-lock-o', 'fa-sign-in', 'fa-sign-out', 'fa-search', 'fa-plus', 'fa-refresh',
|
||
'fa-times', 'fa-check', 'fa-pencil', 'fa-trash', 'fa-play', 'fa-arrow-left',
|
||
'fa-arrow-down', 'fa-arrow-up',
|
||
]
|
||
|
||
/** 会话超时：5 分钟 */
|
||
export const SESSION_TIMEOUT = 5 * 60 * 1000
|
||
|
||
/** 系统监控刷新间隔 */
|
||
export const METRICS_INTERVAL = 30000
|
||
|
||
/** 训练进度刷新间隔 */
|
||
export const TRAINING_REFRESH_INTERVAL = 5000
|