From 4120d605114c1f99b578351f440224b83c0c1989 Mon Sep 17 00:00:00 2001 From: tzwang Date: Fri, 12 Jul 2024 17:58:43 +0800 Subject: [PATCH] updated aioption Former-commit-id: ba671ba0ca3d1c1ae66c229aedccdb7e5ef280d3 --- internal/scheduler/database/aiStorage.go | 2 ++ internal/scheduler/schedulers/option/aiOption.go | 1 + pkg/models/taskaimodel_gen.go | 1 + 3 files changed, 4 insertions(+) diff --git a/internal/scheduler/database/aiStorage.go b/internal/scheduler/database/aiStorage.go index ab188766..54b02c20 100644 --- a/internal/scheduler/database/aiStorage.go +++ b/internal/scheduler/database/aiStorage.go @@ -147,6 +147,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri aiOpt.Replica = inferOpt.Replica aiOpt.AdapterId = inferOpt.AdapterId aiOpt.TaskType = inferOpt.ModelType + aiOpt.ModelName = inferOpt.ModelName aiOpt.StrategyName = inferOpt.Strategy } // 构建主任务结构体 @@ -169,6 +170,7 @@ func (s *AiStorage) SaveAiTask(taskId int64, opt option.Option, adapterName stri Replica: int64(aiOpt.Replica), JobId: jobId, TaskType: aiOpt.TaskType, + ModelName: aiOpt.ModelName, Strategy: aiOpt.StrategyName, Status: status, Msg: msg, diff --git a/internal/scheduler/schedulers/option/aiOption.go b/internal/scheduler/schedulers/option/aiOption.go index b3cc4de0..e6b7a838 100644 --- a/internal/scheduler/schedulers/option/aiOption.go +++ b/internal/scheduler/schedulers/option/aiOption.go @@ -15,6 +15,7 @@ type AiOption struct { ComputeCard string CodeType string AlgorithmName string + ModelName string ImageId string SpecId string diff --git a/pkg/models/taskaimodel_gen.go b/pkg/models/taskaimodel_gen.go index 24473016..94a56845 100644 --- a/pkg/models/taskaimodel_gen.go +++ b/pkg/models/taskaimodel_gen.go @@ -55,6 +55,7 @@ type ( DeletedAt *time.Time `db:"deleted_at"` Card string `db:"card"` InferUrl string `db:"infer_url"` + ModelName string `db:"model_name"` } )