diff --git a/routers/api/v1/repo/modelarts.go b/routers/api/v1/repo/modelarts.go index a1e4e0a65..eb787dc0f 100755 --- a/routers/api/v1/repo/modelarts.go +++ b/routers/api/v1/repo/modelarts.go @@ -471,11 +471,8 @@ func ModelList(ctx *context.APIContext) { var fileInfos []storage.FileInfo if task.ComputeResource == models.NPUResource { //fileInfos, err = storage.GetObsListObject(task.JobName, "output/", parentDir, versionName) - prefix := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, task.JobName, "output/", versionName, parentDir), "/") - if !strings.HasSuffix(prefix, "/") { - prefix += "/" - } - fileInfos, err = storage.GetOneLevelAllObjectUnderDir(setting.Bucket, prefix, "") + prefix := strings.TrimPrefix(path.Join(setting.TrainJobModelPath, task.JobName, setting.OutPutPath, versionName), "/") + fileInfos, err = storage.GetOneLevelAllObjectUnderDir(setting.Bucket, prefix, parentDir) if err != nil { log.Info("get TrainJobListModel failed:", err) ctx.ServerError("GetObsListObject:", err)