From d8c53fd71cdb2463b1e08438670c57c26f01cb13 Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 09:00:35 +0800
Subject: [PATCH 01/11] fix issue

---
 templates/repo/modelarts/inferencejob/show.tmpl | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/templates/repo/modelarts/inferencejob/show.tmpl b/templates/repo/modelarts/inferencejob/show.tmpl
index 78a319747..0128a3e67 100644
--- a/templates/repo/modelarts/inferencejob/show.tmpl
+++ b/templates/repo/modelarts/inferencejob/show.tmpl
@@ -454,7 +454,7 @@ td, th {
                                     class='{{if  and (.CanModify) (eq .Status "KILLED" "FAILED" "START_FAILED" "STOPPED" "COMPLETED") }}ti-download-file{{else}}disabled{{end}}'
                                     href="{{$.RepoLink}}/modelarts/train-job/{{.JobID}}/download_log_file?version_name={{.VersionName}}">
                                     <i class="ri-download-cloud-2-line"></i>
-                                    <span style="margin-left: 0.3rem;">{{$.i18n.Tr "repo.modelarts.download_log"}}</span>{{.CanModify}}
+                                    <span style="margin-left: 0.3rem;">{{$.i18n.Tr "repo.modelarts.download_log"}}</span>
                                 </a>
 
                             </div>
@@ -507,7 +507,6 @@ td, th {
 </div>
 {{template "base/footer" .}}
 <script>
-console.log('{{.CanModify}}')
 $(document).ready(function(){
     $('.secondary.menu .item').tab();
 });

From 42f11142f7423bef99c056ac27a4cce959603152 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Mon, 18 Jul 2022 09:54:02 +0800
Subject: [PATCH 02/11] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/auth/cloudbrain.go        | 1 +
 modules/cloudbrain/cloudbrain.go  | 2 ++
 routers/api/v1/repo/cloudbrain.go | 8 ++++----
 routers/repo/cloudbrain.go        | 2 ++
 4 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/modules/auth/cloudbrain.go b/modules/auth/cloudbrain.go
index 91fa8ad64..160328b5b 100755
--- a/modules/auth/cloudbrain.go
+++ b/modules/auth/cloudbrain.go
@@ -69,6 +69,7 @@ type CreateCloudBrainInferencForm struct {
 	ModelName         string `form:"model_name" binding:"Required"`
 	ModelVersion      string `form:"model_version" binding:"Required"`
 	CkptName          string `form:"ckpt_name" binding:"Required"`
+	LabelName         string `form:"label_names" binding:"Required"`
 }
 
 func (f *CreateCloudBrainForm) Validate(ctx *macaron.Context, errs binding.Errors) binding.Errors {
diff --git a/modules/cloudbrain/cloudbrain.go b/modules/cloudbrain/cloudbrain.go
index dbfff6027..6cbb97999 100755
--- a/modules/cloudbrain/cloudbrain.go
+++ b/modules/cloudbrain/cloudbrain.go
@@ -78,6 +78,7 @@ type GenerateCloudBrainTaskReq struct {
 	ModelName            string
 	ModelVersion         string
 	CkptName             string
+	LabelName            string
 }
 
 func GetCloudbrainDebugCommand() string {
@@ -400,6 +401,7 @@ func GenerateTask(req GenerateCloudBrainTaskReq) error {
 		ModelVersion:         req.ModelVersion,
 		CkptName:             req.CkptName,
 		ResultUrl:            req.ResultPath,
+		LabelName:            req.LabelName,
 		CreatedUnix:          createTime,
 		UpdatedUnix:          createTime,
 		CommitID:             req.CommitID,
diff --git a/routers/api/v1/repo/cloudbrain.go b/routers/api/v1/repo/cloudbrain.go
index 5ebbc9b46..d291024f9 100755
--- a/routers/api/v1/repo/cloudbrain.go
+++ b/routers/api/v1/repo/cloudbrain.go
@@ -157,15 +157,15 @@ func DelCloudBrainJob(ctx *context.APIContext) {
 
 	if errStr != "" {
 		ctx.JSON(http.StatusOK, map[string]interface{}{
-			"message":     ctx.Tr(errStr),
+			"Message":     ctx.Tr(errStr),
 			"VersionName": "1",
-			"code":        1,
+			"Code":        1,
 		})
 	} else {
 		ctx.JSON(http.StatusOK, map[string]interface{}{
-			"message":     "",
+			"Message":     "",
 			"VersionName": "1",
-			"code":        0,
+			"Code":        0,
 		})
 	}
 
diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 20a1fcc07..cdde29ad0 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -380,6 +380,7 @@ func CloudBrainInferenceJobCreate(ctx *context.Context, form auth.CreateCloudBra
 	codePath := setting.JobPath + jobName + cloudbrain.CodeMountPath
 	resourceSpecId := form.ResourceSpecId
 	branchName := form.BranchName
+	labelName := form.LabelName
 	repo := ctx.Repo.Repository
 
 	ckptUrl := setting.Attachment.Minio.RealPath + form.TrainUrl + form.CkptName
@@ -475,6 +476,7 @@ func CloudBrainInferenceJobCreate(ctx *context.Context, form auth.CreateCloudBra
 		ModelVersion:     form.ModelVersion,
 		CkptName:         form.CkptName,
 		TrainUrl:         form.TrainUrl,
+		LabelName:        labelName,
 	}
 
 	err = cloudbrain.GenerateTask(req)

From 8ec83e9638b4448220a5f00075f20d9535aec8e3 Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 10:06:27 +0800
Subject: [PATCH 03/11] fix issue

---
 templates/repo/cloudbrain/inference/show.tmpl   |  2 +-
 templates/repo/cloudbrain/trainjob/show.tmpl    |  2 +-
 templates/repo/grampus/trainjob/show.tmpl       |  2 +-
 templates/repo/modelarts/inferencejob/show.tmpl |  5 +++--
 templates/repo/modelarts/trainjob/show.tmpl     |  2 +-
 web_src/js/features/cloudbrainShow.js           | 11 ++++++-----
 6 files changed, 13 insertions(+), 11 deletions(-)

diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl
index 7653376dd..e3a42cf7a 100644
--- a/templates/repo/cloudbrain/inference/show.tmpl
+++ b/templates/repo/cloudbrain/inference/show.tmpl
@@ -266,7 +266,7 @@
                             onclick="javascript:parseInfo()">{{$.i18n.Tr "repo.cloudbrain.runinfo"}}</a>
                         
                         <a class="item load-model-file" data-tab="four"
-                            data-path="{{$.RepoLink}}/cloudbrain/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                            data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                     </div>
                     <div class="ui tab active" data-tab="first" style="height:400px">
                         <div style="padding-top: 10px;">
diff --git a/templates/repo/cloudbrain/trainjob/show.tmpl b/templates/repo/cloudbrain/trainjob/show.tmpl
index 2bbad2271..62d4f13de 100644
--- a/templates/repo/cloudbrain/trainjob/show.tmpl
+++ b/templates/repo/cloudbrain/trainjob/show.tmpl
@@ -288,7 +288,7 @@
                             onclick="javascript:parseInfo()">{{$.i18n.Tr "repo.cloudbrain.runinfo"}}</a>
                         <a class="item" data-tab="third{{$k}}"
                             onclick="loadLog({{.VersionName}})">{{$.i18n.Tr "repo.modelarts.log"}}</a>
-                        <a class="item load-model-file" data-tab="four{{$k}}" data-path="{{$.RepoLink}}/cloudbrain/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                        <a class="item load-model-file" data-tab="four{{$k}}" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                     </div>
                     <div class="ui tab active" data-tab="first{{$k}}">
                         <div style="padding-top: 10px;">
diff --git a/templates/repo/grampus/trainjob/show.tmpl b/templates/repo/grampus/trainjob/show.tmpl
index 53b20b61a..80c59c28a 100755
--- a/templates/repo/grampus/trainjob/show.tmpl
+++ b/templates/repo/grampus/trainjob/show.tmpl
@@ -298,7 +298,7 @@
                             data-tab="first{{$k}}">{{$.i18n.Tr "repo.modelarts.train_job.config"}}</a>
                         <a class="item" data-tab="second{{$k}}"
                             onclick="loadLog({{.VersionName}})">{{$.i18n.Tr "repo.modelarts.log"}}</a>
-                        <a class="item load-model-file" data-tab="third{{$k}}" data-path="{{$.RepoLink}}/modelarts/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                        <a class="item load-model-file" data-tab="third{{$k}}" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/modelarts/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                     </div>
                     <div class="ui tab active" data-tab="first{{$k}}">
                         <div style="padding-top: 10px;">
diff --git a/templates/repo/modelarts/inferencejob/show.tmpl b/templates/repo/modelarts/inferencejob/show.tmpl
index 0128a3e67..99deee551 100644
--- a/templates/repo/modelarts/inferencejob/show.tmpl
+++ b/templates/repo/modelarts/inferencejob/show.tmpl
@@ -214,7 +214,7 @@ td, th {
 
                             <a class="active item" data-tab="first">{{$.i18n.Tr "repo.modelarts.train_job.config"}}</a>
                             <a class="item log_bottom" data-tab="second" data-version="{{.VersionName}}">{{$.i18n.Tr "repo.modelarts.log"}}</a>
-                            <a class="item load-model-file" data-tab="third" data-path="{{$.RepoLink}}/modelarts/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                            <a class="item load-model-file" data-tab="third" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/modelarts/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                         </div>
 
                         <div class="ui tab active" data-tab="first" style="height:400px">
@@ -451,7 +451,7 @@ td, th {
                         <div class="ui tab" data-tab="second">
                             <div>
                                 <a id="{{.VersionName}}-log-down"
-                                    class='{{if  and (.CanModify) (eq .Status "KILLED" "FAILED" "START_FAILED" "STOPPED" "COMPLETED") }}ti-download-file{{else}}disabled{{end}}'
+                                    class='{{if  and ($.canDownload) (eq .Status "KILLED" "FAILED" "START_FAILED" "STOPPED" "COMPLETED") }}ti-download-file{{else}}disabled{{end}}'
                                     href="{{$.RepoLink}}/modelarts/train-job/{{.JobID}}/download_log_file?version_name={{.VersionName}}">
                                     <i class="ri-download-cloud-2-line"></i>
                                     <span style="margin-left: 0.3rem;">{{$.i18n.Tr "repo.modelarts.download_log"}}</span>
@@ -507,6 +507,7 @@ td, th {
 </div>
 {{template "base/footer" .}}
 <script>
+console.log('{{$.canDownload}}')
 $(document).ready(function(){
     $('.secondary.menu .item').tab();
 });
diff --git a/templates/repo/modelarts/trainjob/show.tmpl b/templates/repo/modelarts/trainjob/show.tmpl
index aa0ab6836..3936aa620 100755
--- a/templates/repo/modelarts/trainjob/show.tmpl
+++ b/templates/repo/modelarts/trainjob/show.tmpl
@@ -327,7 +327,7 @@
                         <a class="item log_bottom" data-tab="second{{$k}}"
                             data-version="{{.VersionName}}">{{$.i18n.Tr "repo.modelarts.log"}}</a>
                         <a class="item metric_chart" data-tab="four{{$k}}" data-version="{{.VersionName}}">资源占用情况</a>
-                        <a class="item load-model-file" data-tab="third{{$k}}" data-path="{{$.RepoLink}}/modelarts/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                        <a class="item load-model-file" data-tab="third{{$k}}" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/modelarts/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                     </div>
                     <div class="ui tab active" data-tab="first{{$k}}">
                         <div style="padding-top: 10px;">
diff --git a/web_src/js/features/cloudbrainShow.js b/web_src/js/features/cloudbrainShow.js
index f665d4fe2..f8f5b06cf 100644
--- a/web_src/js/features/cloudbrainShow.js
+++ b/web_src/js/features/cloudbrainShow.js
@@ -161,7 +161,8 @@ export default async function initCloudrainSow() {
 
   //
   $(".content-pad").on("click", ".load-model-file", function () {
-    console.log("11111111111");
+    let downloadFlag = $(this).data("download-flag");
+    console.log("11111111111", downloadFlag);
     let version_name = $(this).data("version");
     let parents = $(this).data("parents") || "";
     let filename = $(this).data("filename") || "";
@@ -171,7 +172,7 @@ export default async function initCloudrainSow() {
     console.log(url);
     $.get(url, (data) => {
       $(`#dir_list${version_name}`).empty();
-      renderDir(path, data, version_name);
+      renderDir(path, data, version_name, downloadFlag);
       if (init === "init") {
         $(`input[name=model${version_name}]`).val("");
         $(`input[name=modelback${version_name}]`).val(version_name);
@@ -251,7 +252,7 @@ export default async function initCloudrainSow() {
     }
   }
 
-  function renderDir(path, data, version_name) {
+  function renderDir(path, data, version_name, downloadFlag) {
     let html = "";
     html += "<div class='ui grid' style='margin:0;'>";
     html += "<div class='row' style='padding: 0;'>";
@@ -274,8 +275,8 @@ export default async function initCloudrainSow() {
           data.Dirs[i].FileName +
           "</span>";
       } else {
-        if (downlaodFlag) {
-          html += `<a href="${location.href}/result_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}">`;
+        if (downloadFlag) {
+          html += `<a href="${location.href}/result_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}&jobName=${data.task.JobName}">`;
         } else {
           html += `<a class="disabled">`;
         }

From 7fb80fa1931c948ec77f27bf27bf4ac9ec9f5440 Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 10:16:48 +0800
Subject: [PATCH 04/11] fix issue

---
 templates/repo/cloudbrain/inference/show.tmpl | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl
index e3a42cf7a..4b6bf50e2 100644
--- a/templates/repo/cloudbrain/inference/show.tmpl
+++ b/templates/repo/cloudbrain/inference/show.tmpl
@@ -420,9 +420,8 @@
 
                                                         <td class="ti-text-form-content">
                                                             <div class="text-span text-span-w" id="{{.VersionName}}-labels">
-
                                                                 {{if .LabelName}}
-                                                                    {{range $.labelName}}
+                                                                    {{range .LabelName}}
                                                                     <a class="ui label" title="{{.}}">{{.}}</a>
                                                                     {{end}}
                                                                 {{else}}

From 999841b5feedd99878b20af30fafffdc9e1785ec Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Mon, 18 Jul 2022 10:23:41 +0800
Subject: [PATCH 05/11] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routers/repo/cloudbrain.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index cdde29ad0..5c03e26a7 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -832,6 +832,8 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo
 	}
 
 	ctx.Data["task"] = task
+	labelName := strings.Fields(task.LabelName)
+	ctx.Data["LabelName"] = labelName
 	ctx.Data["jobName"] = task.JobName
 	ctx.Data["displayJobName"] = task.DisplayJobName
 	version_list_task := make([]*models.Cloudbrain, 0)

From 638a927d67d5ecf3f491e900a1fb32809e6f035e Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 10:25:21 +0800
Subject: [PATCH 06/11] fix issue

---
 templates/repo/cloudbrain/inference/show.tmpl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl
index 4b6bf50e2..0c4dcf9cd 100644
--- a/templates/repo/cloudbrain/inference/show.tmpl
+++ b/templates/repo/cloudbrain/inference/show.tmpl
@@ -421,7 +421,7 @@
                                                         <td class="ti-text-form-content">
                                                             <div class="text-span text-span-w" id="{{.VersionName}}-labels">
                                                                 {{if .LabelName}}
-                                                                    {{range .LabelName}}
+                                                                    {{range $.labelName}}
                                                                     <a class="ui label" title="{{.}}">{{.}}</a>
                                                                     {{end}}
                                                                 {{else}}

From 094dcb884bc43f2e0e96360e31140911b3f99487 Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 10:31:58 +0800
Subject: [PATCH 07/11] fix issue

---
 templates/repo/cloudbrain/inference/show.tmpl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl
index 0c4dcf9cd..ee0d4c78f 100644
--- a/templates/repo/cloudbrain/inference/show.tmpl
+++ b/templates/repo/cloudbrain/inference/show.tmpl
@@ -543,6 +543,8 @@
 {{template "base/footer" .}}
 
 <script>
+console.log({{$.labelName}})
+console.log({{$.LabelName}})
     $('.menu .item').tab()
 
     $(document).ready(function () {

From 2f7c32816ebeaadc8809a81a4ba04d8af3bb9e5a Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 10:33:05 +0800
Subject: [PATCH 08/11] fix issue

---
 templates/repo/cloudbrain/inference/show.tmpl | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl
index ee0d4c78f..b0f88abc6 100644
--- a/templates/repo/cloudbrain/inference/show.tmpl
+++ b/templates/repo/cloudbrain/inference/show.tmpl
@@ -421,7 +421,7 @@
                                                         <td class="ti-text-form-content">
                                                             <div class="text-span text-span-w" id="{{.VersionName}}-labels">
                                                                 {{if .LabelName}}
-                                                                    {{range $.labelName}}
+                                                                    {{range $.LabelName}}
                                                                     <a class="ui label" title="{{.}}">{{.}}</a>
                                                                     {{end}}
                                                                 {{else}}
@@ -543,8 +543,6 @@
 {{template "base/footer" .}}
 
 <script>
-console.log({{$.labelName}})
-console.log({{$.LabelName}})
     $('.menu .item').tab()
 
     $(document).ready(function () {

From 2f6bf2c7a20c4e2508a74a1c39d87f5fe8b21aeb Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 10:45:48 +0800
Subject: [PATCH 09/11] fix issue

---
 templates/repo/cloudbrain/inference/show.tmpl |  2 +-
 templates/repo/cloudbrain/trainjob/show.tmpl  |  2 +-
 web_src/js/features/cloudbrainShow.js         | 13 +++++++++----
 3 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/templates/repo/cloudbrain/inference/show.tmpl b/templates/repo/cloudbrain/inference/show.tmpl
index b0f88abc6..909cde468 100644
--- a/templates/repo/cloudbrain/inference/show.tmpl
+++ b/templates/repo/cloudbrain/inference/show.tmpl
@@ -266,7 +266,7 @@
                             onclick="javascript:parseInfo()">{{$.i18n.Tr "repo.cloudbrain.runinfo"}}</a>
                         
                         <a class="item load-model-file" data-tab="four"
-                            data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                        data-gpu-flag="true" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/inference-job/{{.JobID}}/result_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                     </div>
                     <div class="ui tab active" data-tab="first" style="height:400px">
                         <div style="padding-top: 10px;">
diff --git a/templates/repo/cloudbrain/trainjob/show.tmpl b/templates/repo/cloudbrain/trainjob/show.tmpl
index 62d4f13de..da7402eb0 100644
--- a/templates/repo/cloudbrain/trainjob/show.tmpl
+++ b/templates/repo/cloudbrain/trainjob/show.tmpl
@@ -288,7 +288,7 @@
                             onclick="javascript:parseInfo()">{{$.i18n.Tr "repo.cloudbrain.runinfo"}}</a>
                         <a class="item" data-tab="third{{$k}}"
                             onclick="loadLog({{.VersionName}})">{{$.i18n.Tr "repo.modelarts.log"}}</a>
-                        <a class="item load-model-file" data-tab="four{{$k}}" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
+                        <a class="item load-model-file" data-tab="four{{$k}}" data-gpu-flag="true" data-download-flag="{{$.canDownload}}" data-path="{{$.RepoLink}}/cloudbrain/train-job/{{.JobID}}/model_list" data-version="{{.VersionName}}" data-parents="" data-filename="" data-init="init" >{{$.i18n.Tr "repo.model_download"}}</a>
                     </div>
                     <div class="ui tab active" data-tab="first{{$k}}">
                         <div style="padding-top: 10px;">
diff --git a/web_src/js/features/cloudbrainShow.js b/web_src/js/features/cloudbrainShow.js
index f8f5b06cf..da24c9335 100644
--- a/web_src/js/features/cloudbrainShow.js
+++ b/web_src/js/features/cloudbrainShow.js
@@ -161,7 +161,8 @@ export default async function initCloudrainSow() {
 
   //
   $(".content-pad").on("click", ".load-model-file", function () {
-    let downloadFlag = $(this).data("download-flag");
+    let downloadFlag = $(this).data("download-flag") || "";
+    let gpuFlag = $(this).data("gpu-flag") || "";
     console.log("11111111111", downloadFlag);
     let version_name = $(this).data("version");
     let parents = $(this).data("parents") || "";
@@ -172,7 +173,7 @@ export default async function initCloudrainSow() {
     console.log(url);
     $.get(url, (data) => {
       $(`#dir_list${version_name}`).empty();
-      renderDir(path, data, version_name, downloadFlag);
+      renderDir(path, data, version_name, downloadFlag, gpuFlag);
       if (init === "init") {
         $(`input[name=model${version_name}]`).val("");
         $(`input[name=modelback${version_name}]`).val(version_name);
@@ -252,7 +253,7 @@ export default async function initCloudrainSow() {
     }
   }
 
-  function renderDir(path, data, version_name, downloadFlag) {
+  function renderDir(path, data, version_name, downloadFlag, gpuFlag) {
     let html = "";
     html += "<div class='ui grid' style='margin:0;'>";
     html += "<div class='row' style='padding: 0;'>";
@@ -276,7 +277,11 @@ export default async function initCloudrainSow() {
           "</span>";
       } else {
         if (downloadFlag) {
-          html += `<a href="${location.href}/result_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}&jobName=${data.task.JobName}">`;
+          if (gpuFlag) {
+            html += `<a href="${location.href}/result_download?version_name=${version_name}&fileName=${data.Dirs[i].FileName}&parentDir=${data.Dirs[i].ParenDir}&jobName=${data.task.JobName}">`;
+          } else {
+            html += `<a href="${location.href}/result_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}">`;
+          }
         } else {
           html += `<a class="disabled">`;
         }

From e2c92a8939d075a11de40968c88b2deda39438ae Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 11:01:42 +0800
Subject: [PATCH 10/11] fix issue

---
 web_src/js/features/cloudbrainShow.js | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/web_src/js/features/cloudbrainShow.js b/web_src/js/features/cloudbrainShow.js
index da24c9335..f0e1c7f3a 100644
--- a/web_src/js/features/cloudbrainShow.js
+++ b/web_src/js/features/cloudbrainShow.js
@@ -278,9 +278,17 @@ export default async function initCloudrainSow() {
       } else {
         if (downloadFlag) {
           if (gpuFlag) {
-            html += `<a href="${location.href}/result_download?version_name=${version_name}&fileName=${data.Dirs[i].FileName}&parentDir=${data.Dirs[i].ParenDir}&jobName=${data.task.JobName}">`;
+            if (path.includes("model_list")) {
+              html += `<a href="${location.href}/download_model?version_name=${version_name}&fileName=${data.Dirs[i].FileName}&parentDir=${data.Dirs[i].ParenDir}&jobName=${data.task.JobName}">`;
+            } else {
+              html += `<a href="${location.href}/result_download?version_name=${version_name}&fileName=${data.Dirs[i].FileName}&parentDir=${data.Dirs[i].ParenDir}&jobName=${data.task.JobName}">`;
+            }
           } else {
-            html += `<a href="${location.href}/result_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}">`;
+            if (path.includes("model_list")) {
+              html += `<a href="${location.href}/model_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}">`;
+            } else {
+              html += `<a href="${location.href}/result_download?version_name=${version_name}&file_name=${data.Dirs[i].FileName}&parent_dir=${data.Dirs[i].ParenDir}">`;
+            }
           }
         } else {
           html += `<a class="disabled">`;

From 0250df25b4d975239b92743e6ff1732f202ed21a Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Mon, 18 Jul 2022 11:06:23 +0800
Subject: [PATCH 11/11] fix issue

---
 web_src/js/features/cloudbrainShow.js | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/web_src/js/features/cloudbrainShow.js b/web_src/js/features/cloudbrainShow.js
index f0e1c7f3a..7be4a0c8e 100644
--- a/web_src/js/features/cloudbrainShow.js
+++ b/web_src/js/features/cloudbrainShow.js
@@ -16,7 +16,6 @@ export default async function initCloudrainSow() {
 
   function logScroll(version_name) {
     let container = document.querySelector(`#log${version_name}`);
-    console.log(container);
     let scrollTop = container.scrollTop;
     let scrollHeight = container.scrollHeight;
     let clientHeight = container.clientHeight;
@@ -163,14 +162,12 @@ export default async function initCloudrainSow() {
   $(".content-pad").on("click", ".load-model-file", function () {
     let downloadFlag = $(this).data("download-flag") || "";
     let gpuFlag = $(this).data("gpu-flag") || "";
-    console.log("11111111111", downloadFlag);
     let version_name = $(this).data("version");
     let parents = $(this).data("parents") || "";
     let filename = $(this).data("filename") || "";
     let init = $(this).data("init") || "";
     let path = $(this).data("path");
     let url = `/api/v1/repos${path}?version_name=${version_name}&parentDir=${parents}`;
-    console.log(url);
     $.get(url, (data) => {
       $(`#dir_list${version_name}`).empty();
       renderDir(path, data, version_name, downloadFlag, gpuFlag);