From c5a1bd11c790854d58ea60a3be5dec2eb187c306 Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Tue, 22 Nov 2022 17:44:01 +0800
Subject: [PATCH] fix issue

---
 templates/repo/grampus/notebook/gpu/new.tmpl |  1 +
 templates/repo/grampus/notebook/npu/new.tmpl | 23 +++++++++++++++++++-
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/templates/repo/grampus/notebook/gpu/new.tmpl b/templates/repo/grampus/notebook/gpu/new.tmpl
index 02a55dfa9..03a145f17 100644
--- a/templates/repo/grampus/notebook/gpu/new.tmpl
+++ b/templates/repo/grampus/notebook/gpu/new.tmpl
@@ -34,6 +34,7 @@
             <form id="form_id" class="ui form" action="{{.Link}}" method="post">
                 {{.CsrfTokenHtml}}
                 <input type="hidden" name='isBranches' value="{{.Branches}}">
+                <input type="hidden" name="type" value="1">
                 <h3 class="ui top attached header">
                     {{.i18n.Tr "repo.cloudbrain.new"}}
                 </h3>
diff --git a/templates/repo/grampus/notebook/npu/new.tmpl b/templates/repo/grampus/notebook/npu/new.tmpl
index e25e43d1f..a1e5d67e6 100644
--- a/templates/repo/grampus/notebook/npu/new.tmpl
+++ b/templates/repo/grampus/notebook/npu/new.tmpl
@@ -58,7 +58,27 @@
                         <label>{{.i18n.Tr "cloudbrain.task_name"}}</label>
                         <input name="display_job_name" id="cloudbrain_job_name" placeholder="任务名称" value="{{.display_job_name}}" tabindex="3" autofocus required maxlength="36" onkeyup="this.value=this.value.replace(/[, ]/g,'')">
                     </div>
-
+                    <div class="inline required field">
+                        <label>{{.i18n.Tr "repo.modelarts.code_version"}}</label>
+                        <select class="ui dropdown width80 left2 {{if not .Branches}}error{{end}}" id="code_version"
+                            name="branch_name">
+                            {{if .branch_name}}
+                            <option name="branch_name" value="{{.branch_name}}">{{.branch_name}}</option>
+                            {{range $k, $v :=.Branches}}
+                            {{ if ne $v $.branch_name }}
+                            <option name="branch_name" value="{{$v}}">{{$v}}</option>
+                            {{end}}
+                            {{end}}
+                            {{else}}
+                            <option name="branch_name" value="{{.branchName}}">{{.branchName}}</option>
+                            {{range $k, $v :=.Branches}}
+                            {{ if ne $v $.branchName }}
+                            <option name="branch_name" value="{{$v}}">{{$v}}</option>
+                            {{end}}
+                            {{end}}
+                            {{end}}
+                        </select>
+                    </div>
                     <div class="inline required field">
                         <label>{{.i18n.Tr "cloudbrain.mirror"}}</label>
                         <select id="cloudbrain_image" class="ui search dropdown" placeholder="{{.i18n.Tr "cloudbrain.choose_mirror"}}" style='width:385px' name="image_id">
@@ -67,6 +87,7 @@
                             {{end}}
                         </select>
                     </div>
+                    
                     <div id="select-multi-dataset">
 
                     </div>