extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-22 08:13:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Automatically use 'Low' when VRAM<4.5GB

Browse Source
This commit is contained in:
JeLuF 2023-05-21 17:42:47 +02:00
parent 566cb55f36
commit 70a3beeaa2
3 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ui/media/js/auto-save.js
View File

@ -169,6 +169,22 @@ function loadSettings() {
}
})
CURRENTLY_LOADING_SETTINGS = false
} else if (localStorage.length < 2) {
// localStorage is too short for OldSettings
// So this is likely the first time Easy Diffusion is running.
// Initialize vram_usage_level based on the available VRAM
function initGPUProfile(event) {
if ( "detail" in event
&& "active" in event.detail
&& "cuda:0" in event.detail.active
&& event.detail.active["cuda:0"].mem_total <4.5 )
{
vramUsageLevelField.value = "low"
vramUsageLevelField.dispatchEvent(new Event("change"))
}
document.removeEventListener("system_info_update", initGPUProfile)
}
document.addEventListener("system_info_update", initGPUProfile)
} else {
CURRENTLY_LOADING_SETTINGS = true
tryLoadOldSettings()

2
ui/media/js/main.js
View File

@ -239,7 +239,7 @@ function setServerStatus(event) {
break
}
if (SD.serverState.devices) {
setDeviceInfo(SD.serverState.devices)
document.dispatchEvent(new CustomEvent("system_info_update", { detail: SD.serverState.devices}))
}
}

4
ui/media/js/parameters.js
View File

@ -586,7 +586,7 @@ async function getSystemInfo() {
$("#use_gpus").val(activeDeviceIds)
}
setDeviceInfo(devices)
document.dispatchEvent(new CustomEvent("system_info_update", { detail: devices}))
setHostInfo(res["hosts"])
let force = false
if (res["enforce_output_dir"] !== undefined) {
@ -657,3 +657,5 @@ saveSettingsBtn.addEventListener("click", function() {
saveSettingsBtn.classList.add("active")
Promise.all([savePromise, asyncDelay(300)]).then(() => saveSettingsBtn.classList.remove("active"))
})
document.addEventListener("system_info_update", (e) => setDeviceInfo(e.detail))