File size: 3,472 Bytes
34097e9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
async function requestModelsHorde() {
//get the models list from url
// https://stablehorde.net/api/v2/status/models
console.log('requestModelsHorde: ')
const full_url = 'https://stablehorde.net/api/v2/status/models'
let request = await fetch(full_url)
let json = await request.json()
console.log('hordes models json:')
console.dir(json)
return json
}
function addHordeModelMenuItem(model_title, model_name) {
// console.log(model_title,model_name)
const menu_item_element = document.createElement('sp-menu-item')
menu_item_element.className = 'mModelMenuItemHorde'
menu_item_element.innerHTML = model_title
menu_item_element.dataset.name = model_name
return menu_item_element
}
async function refreshModelsHorde() {
try {
let g_models_horde = await requestModelsHorde()
// const models_menu_element = document.getElementById('mModelsMenu')
// models_menu_element.value = ""
//(optional): sort the models
g_models_horde.sort(function (a, b) {
return b.count - a.count
})
// g_models_horde = g_models_horde.sort( compareModelCounts );
document.getElementById('mModelsMenuHorde').innerHTML = ''
let model_item_random = addHordeModelMenuItem('Random', 'Random')
// model_item_random.selected = true
document
.getElementById('mModelsMenuHorde')
.appendChild(model_item_random)
for (let model of g_models_horde) {
// console.log(model.name, model.count) //Log
const model_html_tile = `${model.name}: ${model.count}`
const model_item_element = addHordeModelMenuItem(
model_html_tile,
model.name
)
if (model.name === 'stable_diffusion') {
// TODO: refactor this code outside the for loop
// maybe call it in an init function
//selection the stable diffusion model by default
model_item_element.selected = true
}
document
.getElementById('mModelsMenuHorde')
.appendChild(model_item_element)
}
} catch (e) {
console.warn(e)
}
}
function getModelHorde() {
return [...document.getElementsByClassName('mModelMenuItemHorde')].filter(
(e) => e.selected == true
)[0].dataset.name
}
function getScriptArgs() {
const model = getModelHorde()
const b_nsfw = document.getElementById('chUseNSFW').checked
const b_shared_laion = document.getElementById('chUseSharedLaion').checked
let seed_variation = document.getElementById('slSeedVariation').value
seed_variation = parseInt(seed_variation)
const script_args_json = {
model: model,
nsfw: b_nsfw,
shared_laion: b_shared_laion,
seed_variation: seed_variation,
post_processing_1: 'None',
post_processing_2: 'None',
post_processing_3: 'None',
}
const script_args = Object.values(script_args_json)
return script_args
}
document
.getElementById('btnRefreshModelsHorde')
.addEventListener('click', async () => {
await refreshModelsHorde()
})
const script_name = 'Run on Stable Horde'
refreshModelsHorde() //refresh the model when importing the script
module.exports = {
requestModelsHorde,
refreshModelsHorde,
getModelHorde,
getScriptArgs,
script_name,
}
|