Commit c8045c5a authored by SGKoishi's avatar SGKoishi Committed by AUTOMATIC1111
Browse files

The hide_ui_dir_config flag also restrict write attempt to path settings

parent 26a11776
Loading
Loading
Loading
Loading
+10 −0
Original line number Diff line number Diff line
@@ -77,6 +77,16 @@ parser.add_argument("--disable-safe-unpickle", action='store_true', help="disabl


cmd_opts = parser.parse_args()
restricted_opts = [
    "samples_filename_pattern",
    "outdir_samples",
    "outdir_txt2img_samples",
    "outdir_img2img_samples",
    "outdir_extras_samples",
    "outdir_grids",
    "outdir_txt2img_grids",
    "outdir_save",
]

devices.device, devices.device_interrogate, devices.device_gfpgan, devices.device_bsrgan, devices.device_esrgan, devices.device_scunet, devices.device_codeformer = \
(devices.cpu if any(y in cmd_opts.use_cpu for y in [x, 'all']) else devices.get_optimal_device() for x in ['sd', 'interrogate', 'gfpgan', 'bsrgan', 'esrgan', 'scunet', 'codeformer'])
+7 −1
Original line number Diff line number Diff line
@@ -25,7 +25,7 @@ import gradio.routes

from modules import sd_hijack, sd_models
from modules.paths import script_path
from modules.shared import opts, cmd_opts
from modules.shared import opts, cmd_opts, restricted_opts
if cmd_opts.deepdanbooru:
    from modules.deepbooru import get_deepbooru_tags
import modules.shared as shared
@@ -1430,6 +1430,9 @@ Requested path was: {f}
            if comp_args and isinstance(comp_args, dict) and comp_args.get('visible') is False:
                continue

            if cmd_opts.hide_ui_dir_config and key in restricted_opts:
                continue

            oldval = opts.data.get(key, None)
            opts.data[key] = value

@@ -1447,6 +1450,9 @@ Requested path was: {f}
        if not opts.same_type(value, opts.data_labels[key].default):
            return gr.update(visible=True), opts.dumpjson()

        if cmd_opts.hide_ui_dir_config and key in restricted_opts:
            return gr.update(value=oldval), opts.dumpjson()

        oldval = opts.data.get(key, None)
        opts.data[key] = value