lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-24 16:53:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix args

Browse Source
This commit is contained in:
lllyasviel
2024-01-26 14:28:58 -08:00
parent a25cb3aabe
commit f998a26cdb
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
modules/shared_cmd_options.py
View File

@@ -9,10 +9,7 @@ parser = cmd_args.parser
script_loading.preload_extensions(extensions_dir, parser, extension_list=launch.list_extensions(launch.args.ui_settings_file))
script_loading.preload_extensions(extensions_builtin_dir, parser)
if os.environ.get('IGNORE_CMD_ARGS_ERRORS', None) is None:
cmd_opts = parser.parse_args()
else:
cmd_opts, _ = parser.parse_known_args()
cmd_opts, _ = parser.parse_known_args()
cmd_opts.webui_is_non_local = any([cmd_opts.share, cmd_opts.listen, cmd_opts.ngrok, cmd_opts.server_name])
cmd_opts.disable_extension_access = cmd_opts.webui_is_non_local and not cmd_opts.enable_insecure_extension_access

2
modules_forge/initialization.py
View File

@@ -7,7 +7,7 @@ def initialize_forge():
"Use this when you ara on MAC or have more than 20GB VRAM like RTX4096.")
args_parser.args = args_parser.parser.parse_known_args()[0]
args_parser.args.always_offload_from_vram = False # not args_parser.args.disable_offload_from_vram
args_parser.args.always_offload_from_vram = not args_parser.args.disable_offload_from_vram
import ldm_patched.modules.model_management as model_management
import torch