extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-19 17:39:16 +02: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 42f9abdfe31e16160eb003a6da24449028f7e220.
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'], 'gfpgan': ['GFPGANv1.3'],
'realesrgan': ['RealESRGAN_x4plus'], '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'] MODELS_TO_LOAD_ON_START = ['stable-diffusion', 'vae', 'hypernetwork']
known_models = {} 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') f'possible ({max_usage_level}) on this device ({context.device}). Using "{max_usage_level}" instead')
vram_usage_level = max_usage_level vram_usage_level = max_usage_level
if vram_usage_level != context.vram_usage_level: vram_optimizations = VRAM_USAGE_LEVEL_TO_OPTIMIZATIONS[vram_usage_level]
context.vram_usage_level = vram_usage_level
if vram_optimizations != context.vram_optimizations:
context.vram_optimizations = vram_optimizations
return True return True
return False return False