From 212d6f96a946eb6627969eea694f1715a44cc2e8 Mon Sep 17 00:00:00 2001
From: zhoupzh <zhoupzh@pcl.ac.cn>
Date: Thu, 6 Jan 2022 15:30:34 +0800
Subject: [PATCH] fix issue

---
 templates/repo/modelarts/inferencejob/new.tmpl | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/templates/repo/modelarts/inferencejob/new.tmpl b/templates/repo/modelarts/inferencejob/new.tmpl
index 07e992156..0ad8ff14c 100644
--- a/templates/repo/modelarts/inferencejob/new.tmpl
+++ b/templates/repo/modelarts/inferencejob/new.tmpl
@@ -311,15 +311,16 @@
                 loadCheckpointList(dataID).then((res)=>{
                     res.forEach(element => {
                         const ckptSuffix = element.FileName.split(".")
-                        if(!element.IsDir && ckptSuffix[ckptSuffix.length-1]==='ckpt'){
+                        const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7']
+                        if(!element.IsDir && loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length-1])){
                             html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>`
                         }
                         
                     })
                     $('#model_checkpoint').append(html)
                     $("#select_model_checkpoint").removeClass("loading")
-                    const initVersionText = $('#model_checkpoint div.item:last-child').text()
-                    const initVersionValue = $('#model_checkpoint div.item:last-child').data('value')
+                    const initVersionText = $('#model_checkpoint div.item:first-child').text()
+                    const initVersionValue = $('#model_checkpoint div.item:first-child').data('value')
                     $("#select_model_checkpoint").dropdown('set text',initVersionText)
                     $("#select_model_checkpoint").dropdown('set value',initVersionValue,initVersionText,$('#model_name_version div.item:first-child'))
                 })