gameui/ai
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

preview

Browse Source
This commit is contained in:
桜華
2023-07-07 11:20:42 +08:00
parent 7e5443714a
commit cc4d268c29
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
models/Model.go
View File

@@ -32,7 +32,7 @@ type Model struct {
ModelPath string `json:"model_path"` // 模型路徑(實際存放在服務器上的路徑) ModelPath string `json:"model_path"` // 模型路徑(實際存放在服務器上的路徑)
Status string `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error) Status string `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error)
Progress int `json:"progress"` // (0-100) Progress int `json:"progress"` // (0-100)
Image string `json:"image"` // 封面圖片實際地址 Preview string `json:"preview"` // 模型預覽圖片
Hash string `json:"hash"` // 模型哈希值(sha256) Hash string `json:"hash"` // 模型哈希值(sha256)
Epochs int `json:"epochs"` // 訓練步數 Epochs int `json:"epochs"` // 訓練步數
LearningRate float32 `json:"learning_rate"` // 學習率(0.000005) LearningRate float32 `json:"learning_rate"` // 學習率(0.000005)

4
routers/models.go
View File

@@ -248,8 +248,8 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
//go model.Train() //go model.Train()
} }
} }
if model_new.Image != "" && model_new.Image != model.Image { if model_new.Preview != "" && model_new.Preview != model.Preview {
model.Image = model_new.Image model.Preview = model_new.Preview
} }
if model_new.TriggerWords != "" && model_new.TriggerWords != model.TriggerWords { if model_new.TriggerWords != "" && model_new.TriggerWords != model.TriggerWords {
model.TriggerWords = model_new.TriggerWords model.TriggerWords = model_new.TriggerWords