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

40
models/Model.go
View File

@@ -22,26 +22,26 @@ import (
)
type Model struct {
ID int `json:"id" gorm:"primary_key"` // 模型ID
Name string `json:"name"` // 模型名稱
ModelCheckpoint string `json:"model_checkpoint"` // 模型檢查點
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"` // 模型路徑(實際存放在服務器上的路徑)
Status string `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error|public)
Progress int `json:"progress"` // (0-100)
Preview string `json:"preview"` // 模型預覽圖片
Hash string `json:"hash"` // 模型哈希值(sha256)
Epochs int `json:"epochs"` // 訓練步數
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"` // 模型所在服務器(訓練機或推理機)
Stars StarList `json:"stars"` // 模型的收藏者
User User `json:"user" gorm:"-"` // 模型的所有者
ID int `json:"id" gorm:"primary_key"` // 模型ID
Name string `json:"name"` // 模型名稱
ModelCheckpoint string `json:"model_checkpoint"` // 模型檢查點
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"` // 模型路徑(實際存放在服務器上的路徑)
Status string `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error|public)
Progress int `json:"progress"` // (0-100)
Preview string `json:"preview"` // 模型預覽圖片
Hash string `json:"hash"` // 模型哈希值(sha256)
Epochs int `json:"epochs"` // 訓練步數
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"` // 模型所在服務器(訓練機或推理機)
Stars StarList `json:"stars"` // 模型的收藏者
User *User `json:"user" gorm:"foreignKey:UserID;"` // 模型的所有者
CreatedAt time.Time `json:"created_at" gorm:"autoCreateTime"`
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.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 {
image.User = models.User{ID: image.UserID}
db.First(&image.User)
if image.Preview == "" {
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.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Find(&model_list).Count(&listview.Total)
for _, model := range model_list {
model.User = models.User{ID: model.UserID}
db.Take(&model.User)
}
db.Offset((listview.Page - 1) * listview.PageSize).Limit(listview.PageSize).Preload("User").Find(&model_list).Count(&listview.Total)
listview.List = model_list
listview.Next = listview.Page*listview.PageSize < int(listview.Total)
listview.WriteJSON(w)
@@ -376,6 +373,11 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
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 {
log.Println(err)