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 += `
${element.FileName}
`
}
})
$('#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'))
})