soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Update initialization.py

Browse Source
This commit is contained in:
lllyasviel 2024-01-26 16:11:29 -08:00
parent 4a428f954b
commit b91ada8df3
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules_forge/initialization.py
View File

@ -14,8 +14,12 @@ def initialize_forge():
print('User disabled VRAM offload.')
model_management.ALWAYS_VRAM_OFFLOAD = False
elif model_management.total_vram > 20 * 1024:
print('Automatically disable VRAM offload since user have more than 20GB VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = False
if args_parser.args.always_offload_from_vram:
print('User has more than 20GB VRAM, but forced offloading models from VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = True
else:
print('Automatically disable VRAM offload since user has more than 20GB VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = False
else:
print('Always offload models from VRAM.')
model_management.ALWAYS_VRAM_OFFLOAD = True