Browse Source

fix issue

pull/2548/head
Gitea 2 years ago
parent
commit
45bc5d11d5
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      templates/repo/cloudbrain/inference/new.tmpl
  2. +1
    -1
      templates/repo/modelarts/inferencejob/new.tmpl

+ 1
- 1
templates/repo/cloudbrain/inference/new.tmpl View File

@@ -322,7 +322,7 @@
loadCheckpointList(dataID).then((res)=>{ loadCheckpointList(dataID).then((res)=>{
res.forEach(element => { res.forEach(element => {
const ckptSuffix = element.FileName.split(".") const ckptSuffix = element.FileName.split(".")
const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7']
const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7','pdparams','onnx','pbtxt','keras','mlmodel','cfg','pt']
if(!element.IsDir && loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length-1])){ 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>`
} }


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

@@ -339,7 +339,7 @@
loadCheckpointList(dataID).then((res)=>{ loadCheckpointList(dataID).then((res)=>{
res.forEach(element => { res.forEach(element => {
const ckptSuffix = element.FileName.split(".") const ckptSuffix = element.FileName.split(".")
const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7']
const loadCheckpointFile = ['ckpt','pb','h5','json','pkl','pth','t7','pdparams','onnx','pbtxt','keras','mlmodel','cfg','pt']
if(!element.IsDir && loadCheckpointFile.includes(ckptSuffix[ckptSuffix.length-1])){ 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>`
} }


Loading…
Cancel
Save