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

列表附上用户信息

Browse Source
This commit is contained in:
桜華
2023-08-13 06:28:30 +08:00
parent 1b8321349e
commit 06804d96bb
4 changed files with 52 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
models/Image.go
View File

@@ -16,30 +16,30 @@ import (
) )
type Image struct { type Image struct {
ID int `json:"id" gorm:"primary_key"` // ID ID int `json:"id" gorm:"primary_key"` // ID
Name string `json:"name"` // 名称 Name string `json:"name"` // 名称
Hash string `json:"hash"` // 哈希值 Hash string `json:"hash"` // 哈希值
Path string `json:"path"` // 路径 Path string `json:"path"` // 路径
Type string `json:"type"` // 类型 Type string `json:"type"` // 类型
Size int `json:"size"` // 大小 Size int `json:"size"` // 大小
Width int `json:"width"` // 宽度 Width int `json:"width"` // 宽度
Height int `json:"height"` // 高度 Height int `json:"height"` // 高度
Format string `json:"format"` // 格式 Format string `json:"format"` // 格式
Prompt string `json:"prompt"` // 提示词 Prompt string `json:"prompt"` // 提示词
NegativePrompt string `json:"negative_prompt"` // 负向提示 NegativePrompt string `json:"negative_prompt"` // 负向提示
Steps int `json:"steps"` // 迭代步数 (Steps 1~150) Steps int `json:"steps"` // 迭代步数 (Steps 1~150)
CfgScale int `json:"cfg_scale"` // 引导比例(minimum: 1; maximum: 20) CfgScale int `json:"cfg_scale"` // 引导比例(minimum: 1; maximum: 20)
SamplerName string `json:"sampler_name"` // 采样器名称 SamplerName string `json:"sampler_name"` // 采样器名称
Seed int `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647) Seed int `json:"seed"` // 随机种子(minimum: 0; maximum: 2147483647)
FromImage int `json:"from_image"` // 来源图片(如果是从图片生成的, 则记录来源图片的ID) FromImage int `json:"from_image"` // 来源图片(如果是从图片生成的, 则记录来源图片的ID)
Task string `json:"task"` // 任务编号(uuid) Task string `json:"task"` // 任务编号(uuid)
Status string `json:"status"` // 任务状态(queued|running|finished|failed) Status string `json:"status"` // 任务状态(queued|running|finished|failed)
Progress int `json:"progress"` // 任务进度(0-100) Progress int `json:"progress"` // 任务进度(0-100)
Public bool `json:"public"` // 是否公开 Public bool `json:"public"` // 是否公开
UserID int `json:"user_id"` // 用户ID UserID int `json:"user_id"` // 用户ID
ModelID int `json:"model_id"` // 模型ID ModelID int `json:"model_id"` // 模型ID
Preview string `json:"preview" gorm:"-"` // 实时预览 base64 Preview string `json:"preview" gorm:"-"` // 实时预览 base64 或 url
User User `json:"user" gorm:"-"` // 用户 User *User `json:"user" gorm:"foreignKey:UserID;"` // 用户
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"` CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"` UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
} }

40
models/Model.go
View File

@@ -22,26 +22,26 @@ import (
) )
type Model struct { type Model struct {
ID int `json:"id" gorm:"primary_key"` // 模型ID ID int `json:"id" gorm:"primary_key"` // 模型ID
Name string `json:"name"` // 模型名稱 Name string `json:"name"` // 模型名稱
ModelCheckpoint string `json:"model_checkpoint"` // 模型檢查點 ModelCheckpoint string `json:"model_checkpoint"` // 模型檢查點
Info string `json:"info"` // 模型描述 Info string `json:"info"` // 模型描述
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" default:"initial"` // (initial|ready|waiting|running|success|error|public) Status string `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error|public)
Progress int `json:"progress"` // (0-100) Progress int `json:"progress"` // (0-100)
Preview string `json:"preview"` // 模型預覽圖片 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)
Tags TagList `json:"tags"` // 模型標籤(標籤名數組) Tags TagList `json:"tags"` // 模型標籤(標籤名數組)
UserID int `json:"user_id"` // 模型的所有者 UserID int `json:"user_id"` // 模型的所有者
DatasetID int `json:"dataset_id"` // 模型所使用的數據集ID DatasetID int `json:"dataset_id"` // 模型所使用的數據集ID
ServerID string `json:"server_id"` // 模型所在服務器(訓練機或推理機) ServerID string `json:"server_id"` // 模型所在服務器(訓練機或推理機)
Stars StarList `json:"stars"` // 模型的收藏者 Stars StarList `json:"stars"` // 模型的收藏者
User User `json:"user" gorm:"-"` // 模型的所有者 User *User `json:"user" gorm:"foreignKey:UserID;"` // 模型的所有者
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"` CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"` UpdatedAt time.Time `json:"updated_at" gorm:"autoUpdateTime"`
} }

4
routers/images.go
View File

@@ -116,10 +116,8 @@ func ImagesGet(w http.ResponseWriter, r *http.Request) {
db = db.Where("id IN (?)", list) db = db.Where("id IN (?)", list)
} }
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&image_list).Count(&listview.Total) db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Preload("User").Find(&image_list).Count(&listview.Total)
for _, image := range image_list { for _, image := range image_list {
image.User = models.User{ID: image.UserID}
db.First(&image.User)
if image.Preview == "" { if image.Preview == "" {
image.Preview = "https://image.gameuiux.cn/2023/06/27/1687851028u=3116699095,2862677591&fm=253&fmt=auto&app=120&f=JPEG.webp" image.Preview = "https://image.gameuiux.cn/2023/06/27/1687851028u=3116699095,2862677591&fm=253&fmt=auto&app=120&f=JPEG.webp"
} }

12
routers/models.go
View File

@@ -142,11 +142,8 @@ func ModelsGet(w http.ResponseWriter, r *http.Request) {
db = db.Where("id IN (?)", list) db = db.Where("id IN (?)", list)
} }
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&model_list).Count(&listview.Total) db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Preload("User").Find(&model_list).Count(&listview.Total)
for _, model := range model_list {
model.User = models.User{ID: model.UserID}
db.Take(&model.User)
}
listview.List = model_list listview.List = model_list
listview.Next = listview.Page*listview.PageSize < int(listview.Total) listview.Next = listview.Page*listview.PageSize < int(listview.Total)
listview.WriteJSON(w) listview.WriteJSON(w)
@@ -376,6 +373,11 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
model.Tags = model_new.Tags model.Tags = model_new.Tags
} }
// TODO: 只允许管理员更新模型
if model_new.UserID != 0 && model_new.UserID != model.UserID {
model.UserID = model_new.UserID
}
// 執行更新 // 執行更新
if err := configs.ORMDB().Save(&model).Error; err != nil { if err := configs.ORMDB().Save(&model).Error; err != nil {
log.Println(err) log.Println(err)