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

model

Browse Source
This commit is contained in:
桜華
2023-05-13 04:39:47 +08:00
parent 2a9bd12882
commit 9870c903ca
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
models/Model.go
View File

@@ -8,13 +8,13 @@ import (
type Model struct { type Model struct {
ID int `json:"id"` ID int `json:"id"`
Name string `json:"name"` Name string `json:"name"`
Type string `json:"type"` // (lora|ckp|hyper|ti) Type string `json:"type"` // (lora|ckp|hyper|ti)
TriggerWords string `json:"trigger_words"` // 觸發詞 TriggerWords string `json:"trigger_words"` // 觸發詞
BaseModel string `json:"base_model"` // (SD1.5|SD2) BaseModel string `json:"base_model"` // (SD1.5|SD2)
ModelPath string `json:"model_path"` // 模型路徑 ModelPath string `json:"model_path"` // 模型路徑
Status string `json:"status"` // (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"` // 封面圖片實際地址 Image string `json:"image"` // 封面圖片實際地址
Tags string `json:"tags"` Tags string `json:"tags"`
CreatedAt string `json:"created_at"` CreatedAt string `json:"created_at"`
UpdatedAt string `json:"updated_at"` UpdatedAt string `json:"updated_at"`
@@ -51,13 +51,13 @@ func (model *Model) Create() error {
return err return err
} }
defer db.Close() defer db.Close()
stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)") stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, image, tags, created_at, updated_at, user_id) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return err return err
} }
defer stmt.Close() defer stmt.Close()
result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Tags, model.CreatedAt, model.UpdatedAt, model.UserID) result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Image, model.Tags, model.CreatedAt, model.UpdatedAt, model.UserID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return err return err
@@ -119,7 +119,7 @@ func (model *Model) Get() error {
return err return err
} }
defer db.Close() defer db.Close()
err = db.QueryRow("SELECT * FROM models WHERE id = ?", model.ID).Scan(&model.ID, &model.Name, &model.Type, &model.CreatedAt, &model.UpdatedAt) err = db.QueryRow("SELECT id,name,type,trigger_words,base_model,model_path,status,progress,image,tags,creaated_at,updated_at,user_id FROM models WHERE id = ?", model.ID).Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Image, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return err return err
@@ -135,7 +135,7 @@ func QueryModel(id int) (model Model) {
return return
} }
defer db.Close() defer db.Close()
err = db.QueryRow("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id FROM models WHERE id = ?", id).Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID) err = db.QueryRow("SELECT id,name,type,trigger_words,base_model,model_path,status,progress,image,tags,created_at,updated_at,user_id FROM models WHERE id = ?", id).Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Image, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return return
@@ -150,7 +150,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) {
return return
} }
defer db.Close() defer db.Close()
rows, err := db.Query("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize) rows, err := db.Query("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, image, tags, created_at, updated_at, user_id FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return return
@@ -158,7 +158,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) {
defer rows.Close() defer rows.Close()
for rows.Next() { for rows.Next() {
var model Model var model Model
err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID) err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Image, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
if err != nil { if err != nil {
log.Println(err) log.Println(err)
return return