holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Tweak Memory Usage setting text; Fix a bug with the memory usage setting comparison

Browse Source
This commit is contained in:
cmdr2 2022-12-16 14:11:55 +05:30
parent 7982a9ae25
commit 25639cc3f8
2 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -97,8 +97,8 @@ var PARAMETERS = [
id: "vram_usage_level",
type: ParameterType.select,
label: "GPU Memory Usage",
note: "Faster performance requires more GPU memory<br/><br/>" +
"<b>Balanced:</b> almost as fast as High, significantly lower GPU memory usage<br/>" +
note: "Faster performance requires more GPU memory (VRAM)<br/><br/>" +
"<b>Balanced:</b> nearly as fast as High, much lower VRAM usage<br/>" +
"<b>High:</b> fastest, maximum GPU memory usage</br>" +
"<b>Low:</b> slowest, force-used for GPUs with 4 GB (or less) memory",
icon: "fa-forward",

10
ui/sd_internal/model_manager.py
View File

@ -125,19 +125,13 @@ def resolve_model_paths(task_data: TaskData):
if task_data.use_upscale: task_data.use_upscale = resolve_model_to_use(task_data.use_upscale, 'gfpgan')
def set_vram_optimizations(context: Context):
def is_greater(a, b): # is a > b?
if a == "low": # b will be "low", "balanced" or "high"
return False
elif a == "balanced" and b != "low": # b will be "balanced" or "high"
return False
return True
config = app.getConfig()
max_usage_level = device_manager.get_max_vram_usage_level(context.device)
vram_usage_level = config.get('vram_usage_level', 'balanced')
if is_greater(vram_usage_level, max_usage_level):
v = {'low': 0, 'balanced': 1, 'high': 2}
if v[vram_usage_level] > v[max_usage_level]:
log.error(f'Requested GPU Memory Usage level ({vram_usage_level}) is higher than what is ' + \
f'possible ({max_usage_level}) on this device ({context.device}). Using "{max_usage_level}" instead')
vram_usage_level = max_usage_level