學習率

models/Model.go

@@ -13,21 +13,23 @@ import (
 )
 
 type Model struct {
-	ID           int       `json:"id" gorm:"primary_key"`
-	Name         string    `json:"name"`
-	Type         string    `json:"type"`                     // (lora|ckp|hyper|ti)
+	ID           int       `json:"id" gorm:"primary_key"`    // 模型ID
+	Name         string    `json:"name"`                     // 模型名稱
+	Info         string    `json:"info"`                     // 模型描述
+	Type         string    `json:"type"`                     // 模型類型(lora|ckp|hyper|ti)
 	TriggerWords string    `json:"trigger_words"`            // 觸發詞
-	BaseModel    string    `json:"base_model"`               // (SD1.5|SD2)
-	ModelPath    string    `json:"model_path"`               // 模型路徑
+	BaseModel    string    `json:"base_model"`               // 基礎模型(SD1.5|SD2)
+	ModelPath    string    `json:"model_path"`               // 模型路徑(實際存放在服務器上的路徑)
 	Status       string    `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error)
 	Progress     int       `json:"progress"`                 // (0-100)
 	Image        string    `json:"image"`                    // 封面圖片實際地址
 	Hash         string    `json:"hash"`                     // 模型哈希值
 	Epochs       int       `json:"epochs"`                   // 訓練步數
-	Tags         TagList   `json:"tags"`
-	UserID       int       `json:"user_id"`
-	DatasetID    int       `json:"dataset_id"`
-	ServerID     string    `json:"server_id"`
+	LearningRate float32   `json:"learning_rate"`            // 學習率(0.000005)
+	Tags         TagList   `json:"tags"`                     // 模型標籤(標籤名數組)
+	UserID       int       `json:"user_id"`                  // 模型的所有者
+	DatasetID    int       `json:"dataset_id"`               // 模型所使用的數據集ID
+	ServerID     string    `json:"server_id"`                // 模型所在服務器(訓練機或推理機)
 	CreatedAt    time.Time `json:"created_at" gorm:"autoCreateTime"`
 	UpdatedAt    time.Time `json:"updated_at" gorm:"autoUpdateTime"`
 }