From cc4d268c29655e20e1e4ef54fa4ebcc3f26745c6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= Date: Fri, 7 Jul 2023 11:20:42 +0800 Subject: [PATCH] preview --- models/Model.go | 2 +- routers/models.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/models/Model.go b/models/Model.go index 1f9068c..7a5d35e 100644 --- a/models/Model.go +++ b/models/Model.go @@ -32,7 +32,7 @@ type Model struct { 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"` // 封面圖片實際地址 + Preview string `json:"preview"` // 模型預覽圖片 Hash string `json:"hash"` // 模型哈希值(sha256) Epochs int `json:"epochs"` // 訓練步數 LearningRate float32 `json:"learning_rate"` // 學習率(0.000005) diff --git a/routers/models.go b/routers/models.go index 17f3a42..03b57a4 100644 --- a/routers/models.go +++ b/routers/models.go @@ -248,8 +248,8 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) { //go model.Train() } } - if model_new.Image != "" && model_new.Image != model.Image { - model.Image = model_new.Image + if model_new.Preview != "" && model_new.Preview != model.Preview { + model.Preview = model_new.Preview } if model_new.TriggerWords != "" && model_new.TriggerWords != model.TriggerWords { model.TriggerWords = model_new.TriggerWords