diff --git a/templates/repo/modelsafety/new.tmpl b/templates/repo/modelsafety/new.tmpl
index 83b768ef1..31525b27c 100644
--- a/templates/repo/modelsafety/new.tmpl
+++ b/templates/repo/modelsafety/new.tmpl
@@ -14,6 +14,10 @@
         font-size: 14px !important;
         margin-bottom: 2rem !important;
     }
+    
+    .width100 {
+        width: 100% !important;
+    }
 
     .width81 {
         margin-left: 1.5rem;
@@ -141,7 +145,42 @@
                             <i class="question circle icon" data-content="{{.i18n.Tr "cloudbrain.model_file_postfix_rule"}}" data-position="top center" data-variation="inverted mini"></i>
                         </span>
                     </div>
+
+                    {{if eq .datasetType 0}}
                     <div id="images-new-cb"></div>
+                    {{else}}
+                    <div class="required inline min_title fields" style="width: 95%;">
+                        <label class="label-fix-width" style="font-weight: normal;">{{.i18n.Tr "repo.modelarts.train_job.AI_driver"}}</label>
+                        <div class="field" style="flex: 1.5;">
+                            <select class="ui dropdown width100" id="trainjob_engines">
+                                {{range .engines}}
+                                <option value="{{.Value}}">{{.Value}}</option>
+                                {{end}}
+                            </select>
+                        </div>
+                        <div class="field" style="flex: 2;" id="engine_name">
+                            <select class="ui dropdown width100" id="trainjob_engine_versions" name="engine_id">
+                                {{if .engine_id}}
+                                    {{range .engine_versions}}
+                                        {{if eq $.engine_id .ID}}
+                                            <option value="{{.ID}}">{{.Value}}</option>
+                                        {{end}}
+                                    {{end}}
+                                    {{range .engine_versions}}
+                                        {{if ne $.engine_id .ID}}
+                                            <option value="{{.ID}}">{{.Value}}</option>
+                                        {{end}}
+                                    {{end}}
+                                {{else}}
+                                    {{range .engine_versions}}
+                                        <option value="{{.ID}}">{{.Value}}</option>
+                                    {{end}}
+                                {{end}}
+                            </select>
+                        </div>
+                    </div>
+                    {{end}}
+
                     <div class="inline field min_title required">
                         <label class="label-fix-width" style="font-weight: normal;">{{.i18n.Tr "repo.modelarts.train_job.start_file"}}</label>
                         <input style="width: 48.5%;" name="boot_file" id="trainjob_boot_file" value="{{.boot_file}}" tabindex="3" autofocus required maxlength="255">