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')) })