feat: 扩展类型定义支撑量化、版本与监控
FineTuneTask 新增量化参数字段(QLoRA 位数、量化方法、位数、分组、GGUF 格式),新增 DatasetVersion 与 DatasetVersionList 支撑数据集版本管理,新增 GpuProcess 完善硬件监控。
This commit is contained in:
@@ -80,6 +80,18 @@ export interface DatasetItem {
|
||||
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 TrainMethodType = 'lora' | 'full'
|
||||
@@ -109,6 +121,18 @@ export interface FineTuneTask {
|
||||
lora_alpha?: number
|
||||
lora_dropout?: 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
|
||||
process_id?: number
|
||||
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 {
|
||||
name: string
|
||||
gpu_percent: number
|
||||
@@ -236,16 +268,26 @@ export interface GpuInfo {
|
||||
memory_total_gb: number
|
||||
temperature: 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
|
||||
clock_mhz?: number
|
||||
driver_version?: string
|
||||
}
|
||||
|
||||
export interface SystemInfo {
|
||||
timestamp?: string
|
||||
cpu?: {
|
||||
percent: number
|
||||
cores: number
|
||||
percents?: number[]
|
||||
model?: string
|
||||
frequency_mhz?: number
|
||||
load_1m?: number
|
||||
}
|
||||
memory?: {
|
||||
used_gb: number
|
||||
@@ -258,11 +300,15 @@ export interface SystemInfo {
|
||||
used_gb: number
|
||||
total_gb: number
|
||||
percent: number
|
||||
read_mb_s?: number
|
||||
write_mb_s?: number
|
||||
}
|
||||
gpu?: GpuInfo[]
|
||||
network?: {
|
||||
download_mb?: number
|
||||
upload_mb?: number
|
||||
download_mb_s?: number
|
||||
upload_mb_s?: number
|
||||
}
|
||||
system?: {
|
||||
uptime_seconds?: number
|
||||
|
||||
Reference in New Issue
Block a user