diff --git a/templates/repo/cloudbrain/inference/new.tmpl b/templates/repo/cloudbrain/inference/new.tmpl index b19de1bc5..1e82cf0b9 100644 --- a/templates/repo/cloudbrain/inference/new.tmpl +++ b/templates/repo/cloudbrain/inference/new.tmpl @@ -297,36 +297,42 @@ let nameMap,nameList let faildModelName = $('#failed_model_name').val() // 获取模型列表和模型名称对应的模型版本 - $.get(`${RepoLink}/modelmanage/query_model_for_predict?type=0`, (data) => { + $(document).ready(function(){ modelVersion() modelCkpt() - nameMap = data.nameMap - nameList = data.nameList - let html = '' - nameList.forEach(element => { - html += `
${element}
` - }); - if(nameList.length!==0){ - const initModelVersion = nameMap[nameList[0]][0] - const initTrainTaskInfo = JSON.parse(initModelVersion.TrainTaskInfo) - $('#model_name').append(html) - if(faildModelName){ - $("#select_model").dropdown('set text',faildModelName) - $("#select_model").dropdown('set value',faildModelName,faildModelName) - }else{ - $("#select_model").dropdown('set text',nameList[0]) - $("#select_model").dropdown('set value',nameList[0],nameList[0]) + $.get(`${RepoLink}/modelmanage/query_model_for_predict?type=0`, (data) => { + nameMap = data.nameMap + nameList = data.nameList + let html = '' + nameList.forEach(element => { + html += `
${element}
` + }); + if(nameList.length!==0){ + const initModelVersion = nameMap[nameList[0]][0] + const initTrainTaskInfo = JSON.parse(initModelVersion.TrainTaskInfo) + $('#model_name').append(html) + if(faildModelName){ + $("#select_model").dropdown('set text',faildModelName) + $("#select_model").dropdown('set value',faildModelName,faildModelName) + }else{ + $("#select_model").dropdown('set text',nameList[0]) + $("#select_model").dropdown('set value',nameList[0],nameList[0]) + } } - } - $('#select_model').removeClass("loading") + $('#select_model').removeClass("loading") + }) }) + // 根据选中的模型名称获取相应的模型版本 function modelVersion(){ let faildModelVersion = $('#failed_model_version').val() let faildTrainUrl = $('#failed_train_url').val() $('#select_model').dropdown({ onChange: function(value, text, $selectedItem) { + faildModelVersion = '' + faildTrainUrl='' + console.log("-----------------") $("#select_model_version").addClass("loading") $('#model_name_version').empty() let html = '' @@ -339,6 +345,7 @@ $("#select_model_version").removeClass("loading") const initVersionText = $('#model_name_version div.item:first-child').text() const initVersionValue = $('#model_name_version div.item:first-child').data('value') + if(faildModelVersion&&faildTrainUrl){ console.log("-0-0-0-0-0-") $("#select_model_version").dropdown('set text',faildModelVersion) @@ -347,6 +354,7 @@ $("#select_model_version").dropdown('set text',initVersionText) $("#select_model_version").dropdown('set value',initVersionValue,initVersionText,$('#model_name_version div.item:first-child')) } + } }) } @@ -355,6 +363,8 @@ let faildCkptName = $('#failed_ckpt_name').val() $('#select_model_version').dropdown({ onChange: function(value, text, $selectedItem) { + faildCkptName='' + console.log("===========----------") const dataID = $selectedItem[0].getAttribute("data-id") const label = $selectedItem[0].getAttribute("data-label") $("#select_model_checkpoint").addClass("loading") @@ -373,6 +383,7 @@ $("#select_model_checkpoint").removeClass("loading") const initVersionText = $('#model_checkpoint div.item:first-child').text() const initVersionValue = $('#model_checkpoint div.item:first-child').data('value') + if(faildCkptName){ $("#select_model_checkpoint").dropdown('set text',faildCkptName) $("#select_model_checkpoint").dropdown('set value',faildCkptName,faildCkptName,$('#model_name_version div.item:first-child'))