feat: 扩展类型定义支撑量化、版本与监控
FineTuneTask 新增量化参数字段(QLoRA 位数、量化方法、位数、分组、GGUF 格式),新增 DatasetVersion 与 DatasetVersionList 支撑数据集版本管理,新增 GpuProcess 完善硬件监控。
This commit is contained in:
@@ -80,6 +80,18 @@ export interface DatasetItem {
|
|||||||
files?: DatasetFile[]
|
files?: DatasetFile[]
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export interface DatasetVersion {
|
||||||
|
id: string
|
||||||
|
version: number
|
||||||
|
create_time: string
|
||||||
|
description?: string
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface DatasetVersionList {
|
||||||
|
versions: DatasetVersion[]
|
||||||
|
active_version_id: string
|
||||||
|
}
|
||||||
|
|
||||||
// ============ 训练任务 ============
|
// ============ 训练任务 ============
|
||||||
export type TrainType = 'SFT' | 'DPO' | 'CPT'
|
export type TrainType = 'SFT' | 'DPO' | 'CPT'
|
||||||
export type TrainMethodType = 'lora' | 'full'
|
export type TrainMethodType = 'lora' | 'full'
|
||||||
@@ -109,6 +121,18 @@ export interface FineTuneTask {
|
|||||||
lora_alpha?: number
|
lora_alpha?: number
|
||||||
lora_dropout?: number
|
lora_dropout?: number
|
||||||
lora_rank?: number
|
lora_rank?: number
|
||||||
|
/** 训练时量化位数(QLoRA):0=不量化 / 8 / 4 */
|
||||||
|
quantization_bit?: number
|
||||||
|
/** 训练完成后是否导出量化模型 */
|
||||||
|
export_quantized?: boolean
|
||||||
|
/** 导出量化方法:bnb / gptq / awq / gguf */
|
||||||
|
quant_method?: 'bnb' | 'gptq' | 'awq' | 'gguf' | string
|
||||||
|
/** 导出量化位数 */
|
||||||
|
quant_bits?: number
|
||||||
|
/** 分组大小(GPTQ / AWQ) */
|
||||||
|
quant_group_size?: number
|
||||||
|
/** GGUF 导出格式 */
|
||||||
|
export_format?: string
|
||||||
progress?: number
|
progress?: number
|
||||||
process_id?: number
|
process_id?: number
|
||||||
train_duration?: string
|
train_duration?: string
|
||||||
@@ -229,6 +253,14 @@ export interface Dimension {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// ============ 系统监控 ============
|
// ============ 系统监控 ============
|
||||||
|
export interface GpuProcess {
|
||||||
|
pid: number
|
||||||
|
name: string
|
||||||
|
memory_used_gb: number
|
||||||
|
task_name?: string
|
||||||
|
user?: string
|
||||||
|
}
|
||||||
|
|
||||||
export interface GpuInfo {
|
export interface GpuInfo {
|
||||||
name: string
|
name: string
|
||||||
gpu_percent: number
|
gpu_percent: number
|
||||||
@@ -236,16 +268,26 @@ export interface GpuInfo {
|
|||||||
memory_total_gb: number
|
memory_total_gb: number
|
||||||
temperature: number
|
temperature: number
|
||||||
power_w: number
|
power_w: number
|
||||||
|
id?: number
|
||||||
|
uuid?: string
|
||||||
|
status?: 'idle' | 'busy' | 'warning' | 'offline'
|
||||||
|
memory_percent?: number
|
||||||
|
power_limit_w?: number
|
||||||
|
processes?: GpuProcess[]
|
||||||
fan_speed?: number
|
fan_speed?: number
|
||||||
clock_mhz?: number
|
clock_mhz?: number
|
||||||
driver_version?: string
|
driver_version?: string
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface SystemInfo {
|
export interface SystemInfo {
|
||||||
|
timestamp?: string
|
||||||
cpu?: {
|
cpu?: {
|
||||||
percent: number
|
percent: number
|
||||||
cores: number
|
cores: number
|
||||||
percents?: number[]
|
percents?: number[]
|
||||||
|
model?: string
|
||||||
|
frequency_mhz?: number
|
||||||
|
load_1m?: number
|
||||||
}
|
}
|
||||||
memory?: {
|
memory?: {
|
||||||
used_gb: number
|
used_gb: number
|
||||||
@@ -258,11 +300,15 @@ export interface SystemInfo {
|
|||||||
used_gb: number
|
used_gb: number
|
||||||
total_gb: number
|
total_gb: number
|
||||||
percent: number
|
percent: number
|
||||||
|
read_mb_s?: number
|
||||||
|
write_mb_s?: number
|
||||||
}
|
}
|
||||||
gpu?: GpuInfo[]
|
gpu?: GpuInfo[]
|
||||||
network?: {
|
network?: {
|
||||||
download_mb?: number
|
download_mb?: number
|
||||||
upload_mb?: number
|
upload_mb?: number
|
||||||
|
download_mb_s?: number
|
||||||
|
upload_mb_s?: number
|
||||||
}
|
}
|
||||||
system?: {
|
system?: {
|
||||||
uptime_seconds?: number
|
uptime_seconds?: number
|
||||||
|
|||||||
Reference in New Issue
Block a user