diff --git a/public/scripts/samplerSelect.js b/public/scripts/samplerSelect.js index 0389abd6c..687bbe0a3 100644 --- a/public/scripts/samplerSelect.js +++ b/public/scripts/samplerSelect.js @@ -90,6 +90,11 @@ function setSamplerListListeners() { targetDisplayType = 'block'; } + if (samplerName === 'xtc_probability') { + relatedDOMElement = $('#xtc_block'); + targetDisplayType = 'block'; + } + if (samplerName === 'dynatemp') { relatedDOMElement = $('#dynatemp_block_ooba'); targetDisplayType = 'block'; @@ -248,6 +253,10 @@ async function listSamplers(main_api, arrayOnly = false) { targetDOMelement = $('#dryBlock'); displayname = 'DRY Rep Pen Block'; } + if (sampler === 'xtc_probability') { + targetDOMelement = $('#xtc_block'); + displayname = 'XTC Block'; + } if (sampler === 'dynatemp') { targetDOMelement = $('#dynatemp_block_ooba'); @@ -374,6 +383,11 @@ export async function validateDisabledSamplers(redraw = false) { targetDisplayType = 'block'; } + if (sampler === 'xtc_probability') { + relatedDOMElement = $('#xtc_block'); + targetDisplayType = 'block'; + } + if (sampler === 'penalty_alpha') { //contrastive search only has one sampler, does it need its own block? relatedDOMElement = $('#contrastiveSearchBlock'); }