floraachy
/
pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated aioption 

Former-commit-id: ba671ba0ca3d1c1ae66c229aedccdb7e5ef280d3
This commit is contained in:
tzwang 2024-07-12 17:58:43 +08:00
parent a5e6313974
commit 4120d60511
3 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/scheduler/database/aiStorage.go
View File

@ -147,6 +147,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri
aiOpt.Replica = inferOpt.Replica aiOpt.Replica = inferOpt.Replica
aiOpt.AdapterId = inferOpt.AdapterId aiOpt.AdapterId = inferOpt.AdapterId
aiOpt.TaskType = inferOpt.ModelType aiOpt.TaskType = inferOpt.ModelType
aiOpt.ModelName = inferOpt.ModelName
aiOpt.StrategyName = inferOpt.Strategy aiOpt.StrategyName = inferOpt.Strategy
} }
// 构建主任务结构体 // 构建主任务结构体
@ -169,6 +170,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri
Replica: int64(aiOpt.Replica), Replica: int64(aiOpt.Replica),
JobId: jobId, JobId: jobId,
TaskType: aiOpt.TaskType, TaskType: aiOpt.TaskType,
ModelName: aiOpt.ModelName,
Strategy: aiOpt.StrategyName, Strategy: aiOpt.StrategyName,
Status: status, Status: status,
Msg: msg, Msg: msg,

1
internal/scheduler/schedulers/option/aiOption.go
View File

@ -15,6 +15,7 @@ type AiOption struct {
ComputeCard string ComputeCard string
CodeType string CodeType string
AlgorithmName string AlgorithmName string
ModelName string
ImageId string ImageId string
SpecId string SpecId string

1
pkg/models/taskaimodel_gen.go
View File

@ -55,6 +55,7 @@ type (
DeletedAt *time.Time `db:"deleted_at"` DeletedAt *time.Time `db:"deleted_at"`
Card string `db:"card"` Card string `db:"card"`
InferUrl string `db:"infer_url"` InferUrl string `db:"infer_url"`
ModelName string `db:"model_name"`
} }
) )