extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-24 14:22:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Don't set the specific vram optimizations to use, instead use the new sdkit API for setting the vram usage level directly"

Browse Source 
This reverts commit 42f9abdfe3.
This commit is contained in:
cmdr2 2023-01-18 10:30:56 +05:30
parent 9b1a9cc7c8
commit 52458ae273
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ui/easydiffusion/model_manager.py
View File

@ -24,6 +24,11 @@ DEFAULT_MODELS = {
'gfpgan': ['GFPGANv1.3'],
'realesrgan': ['RealESRGAN_x4plus'],
}
VRAM_USAGE_LEVEL_TO_OPTIMIZATIONS = {
'balanced': {'KEEP_FS_AND_CS_IN_CPU', 'SET_ATTENTION_STEP_TO_4'},
'low': {'KEEP_ENTIRE_MODEL_IN_CPU'},
'high': {},
}
MODELS_TO_LOAD_ON_START = ['stable-diffusion', 'vae', 'hypernetwork']
known_models = {}
@ -128,8 +133,10 @@ def set_vram_optimizations(context: Context):
f'possible ({max_usage_level}) on this device ({context.device}). Using "{max_usage_level}" instead')
vram_usage_level = max_usage_level
if vram_usage_level != context.vram_usage_level:
context.vram_usage_level = vram_usage_level
vram_optimizations = VRAM_USAGE_LEVEL_TO_OPTIMIZATIONS[vram_usage_level]
if vram_optimizations != context.vram_optimizations:
context.vram_optimizations = vram_optimizations
return True
return False