Browse Source

Merge remote-tracking branch 'gitlink/master'

pull/538/head
jagger 2 months ago
parent
commit
ff72602018
2 changed files with 8 additions and 6 deletions
  1. +7
    -5
      internal/scheduler/service/aiService.go
  2. +1
    -1
      internal/storeLink/modelarts.go

+ 7
- 5
internal/scheduler/service/aiService.go View File

@@ -41,11 +41,13 @@ type AiService struct {
} }


func NewAiService(conf *config.Config, storages *database.AiStorage, localCache map[string]interface{}) (*AiService, error) { 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{ aiService := &AiService{
AiExecutorAdapterMap: make(map[string]map[string]executor.AiExecutor), AiExecutorAdapterMap: make(map[string]map[string]executor.AiExecutor),
AiCollectorAdapterMap: make(map[string]map[string]collector.AiCollector), AiCollectorAdapterMap: make(map[string]map[string]collector.AiCollector),


+ 1
- 1
internal/storeLink/modelarts.go View File

@@ -183,7 +183,7 @@ func (m *ModelArtsLink) SubmitTask(ctx context.Context, imageId string, cmd stri
} }
if len(datasetsId) != 0 { if len(datasetsId) != 0 {
inputs = append(inputs, &modelarts.InputTraining{ inputs = append(inputs, &modelarts.InputTraining{
Name: "input",
Name: "dataset_input",
AccessMethod: "parameter", AccessMethod: "parameter",
Remote: &modelarts.RemoteTra{ Remote: &modelarts.RemoteTra{
Obs: &modelarts.ObsTra{ Obs: &modelarts.ObsTra{


Loading…
Cancel
Save