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"`
 	}
 )