lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 11:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update initialization.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 22:52:11 -08:00
parent ea1f3ba2e4
commit cb34e10956
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
modules_forge/initialization.py
View File

@@ -1,5 +1,18 @@
import sys
def initialize_forge(): def initialize_forge():
bad_list = ['--lowvram', '--medvram', '--medvram-sdxl']
for bad in bad_list:
if bad in sys.argv:
print(f'Arg {bad} is removed in Forge.')
print(f'Now memory management is fully automatic and you do not need any command flags.')
print(f'Please just remove this flag.')
print(f'In extreme cases, if you want to force previous lowvram/medvram behaviors, '
f'please use --always-offload-from-vram')
exit(0)
from ldm_patched.modules import args_parser from ldm_patched.modules import args_parser
args_parser.args, _ = args_parser.parser.parse_known_args() args_parser.args, _ = args_parser.parser.parse_known_args()