Browse Source

fix issue

pull/1310/head
zhoupzh 3 years ago
parent
commit
212d6f96a9
1 changed files with 4 additions and 3 deletions
  1. +4
    -3
      templates/repo/modelarts/inferencejob/new.tmpl

+ 4
- 3
templates/repo/modelarts/inferencejob/new.tmpl View File

@@ -311,15 +311,16 @@
loadCheckpointList(dataID).then((res)=>{ loadCheckpointList(dataID).then((res)=>{
res.forEach(element => { res.forEach(element => {
const ckptSuffix = element.FileName.split(".") 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>` html += `<div class="item" data-value=${element.FileName}>${element.FileName}</div>`
} }
}) })
$('#model_checkpoint').append(html) $('#model_checkpoint').append(html)
$("#select_model_checkpoint").removeClass("loading") $("#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 text',initVersionText)
$("#select_model_checkpoint").dropdown('set value',initVersionValue,initVersionText,$('#model_name_version div.item:first-child')) $("#select_model_checkpoint").dropdown('set value',initVersionValue,initVersionText,$('#model_name_version div.item:first-child'))
}) })


Loading…
Cancel
Save