diff --git a/internal/scheduler/service/aiService.go b/internal/scheduler/service/aiService.go index 5d5af117..254f19ba 100644 --- a/internal/scheduler/service/aiService.go +++ b/internal/scheduler/service/aiService.go @@ -41,11 +41,13 @@ type AiService struct { } func NewAiService(conf *config.Config, storages *database.AiStorage, localCache map[string]interface{}) (*AiService, error) { - var aiType = "1" - adapterIds, err := storages.GetAdapterIdsByType(aiType) - if err != nil { - return nil, err - } + //var aiType = "1" + var tempAdapterId = "1777144940459986944" + adapterIds := []string{tempAdapterId} + //adapterIds, err := storages.GetAdapterIdsByType(aiType) + //if err != nil { + // return nil, err + //} aiService := &AiService{ AiExecutorAdapterMap: make(map[string]map[string]executor.AiExecutor), AiCollectorAdapterMap: make(map[string]map[string]collector.AiCollector), diff --git a/internal/storeLink/modelarts.go b/internal/storeLink/modelarts.go index af3c517c..99aa5d53 100644 --- a/internal/storeLink/modelarts.go +++ b/internal/storeLink/modelarts.go @@ -183,7 +183,7 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri } if len(datasetsId) != 0 { inputs = append(inputs, &modelarts.InputTraining{ - Name: "input", + Name: "dataset_input", AccessMethod: "parameter", Remote: &modelarts.RemoteTra{ Obs: &modelarts.ObsTra{