From cb40519aed4ac8c6281bf68ff5565c450fccf628 Mon Sep 17 00:00:00 2001
From: zouap <zouap@pcl.ac.cn>
Date: Thu, 13 Oct 2022 16:28:37 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E6=96=B0=E5=A2=9Emodelart?=
 =?UTF-8?q?=E5=BC=95=E6=93=8E=E5=90=8E=E7=9A=84=E9=97=AE=E9=A2=98=E3=80=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: zouap <zouap@pcl.ac.cn>
---
 templates/repo/modelarts/trainjob/show.tmpl  | 4 ++--
 templates/repo/modelmanage/convertIndex.tmpl | 4 ++--
 templates/repo/modelmanage/showinfo.tmpl     | 4 ++--
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/templates/repo/modelarts/trainjob/show.tmpl b/templates/repo/modelarts/trainjob/show.tmpl
index 1cdb595dd..e95a1233d 100755
--- a/templates/repo/modelarts/trainjob/show.tmpl
+++ b/templates/repo/modelarts/trainjob/show.tmpl
@@ -892,11 +892,11 @@
                     $('#JobName').val(obj.DisplayJobName).addClass('model_disabled')
                     $('input[name="JobId"]').val(obj.JobID)
                     $('input[name="VersionName"]').val(obj.VersionName).addClass('model_disabled')
-                    if(obj.EngineID ==122 || obj.EngineID ==35 || obj.EngineID ==-1){
+                    if(obj.EngineID ==122 || obj.EngineID ==35 || obj.EngineID ==-1 || obj.EngineID ==37){
                         $('input[name="Engine_name"]').val("MindSpore").addClass('model_disabled');
                         $('input[name="Engine"]').val(2);
                     }
-                    if(obj.EngineID ==121){
+                    if(obj.EngineID ==121 || obj.EngineID ==38){
                         $('input[name="Engine_name"]').val("TensorFlow").addClass('model_disabled');
                         $('input[name="Engine"]').val(1);
                     }
diff --git a/templates/repo/modelmanage/convertIndex.tmpl b/templates/repo/modelmanage/convertIndex.tmpl
index 3c4ac7f69..92eefca2e 100644
--- a/templates/repo/modelmanage/convertIndex.tmpl
+++ b/templates/repo/modelmanage/convertIndex.tmpl
@@ -581,10 +581,10 @@
             if(versionList[0].Engine == engineOption){
                 return "selected=\"selected\"";
             }else{
-                if(versionList[0].Engine==122 && engineOption==2){
+                if((versionList[0].Engine==122 || versionList[0].Engine==37) && engineOption==2){
                     return "selected=\"selected\"";
                 }
-                if(versionList[0].Engine==121 && engineOption==1){
+                if((versionList[0].Engine==121 || versionList[0].Engine==38) && engineOption==1){
                     return "selected=\"selected\"";
                 }
             }
diff --git a/templates/repo/modelmanage/showinfo.tmpl b/templates/repo/modelmanage/showinfo.tmpl
index e18c6bd99..4cc0305ed 100644
--- a/templates/repo/modelmanage/showinfo.tmpl
+++ b/templates/repo/modelmanage/showinfo.tmpl
@@ -244,9 +244,9 @@ function loadInfo(){
  function getEngineName(model){
         if(model.Engine == 0){
             return "PyTorch";
-        }else  if(model.Engine == 1 || model.Engine == 121){
+        }else  if(model.Engine == 1 || model.Engine == 121 || model.Engine == 38){
             return "TensorFlow";
-        }else  if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35){
+        }else  if(model.Engine == 2 || model.Engine == 122 || model.Engine == 35 || model.Engine == 37){
             return "MindSpore";
         }else  if(model.Engine == 3){
             return "Other";