diff options
| author | yfszzx <yfszzx@gmail.com> | 2022-10-17 15:23:32 +0800 |
|---|---|---|
| committer | yfszzx <yfszzx@gmail.com> | 2022-10-17 15:23:32 +0800 |
| commit | 2a3e7ed872dc9b8da27cccc7f78df092f4a2f578 (patch) | |
| tree | 5a288fe646eb58d35f4079b734a6d3765a98fd3f /modules/shared.py | |
| parent | 5b1394bead93e5485ced5de10f1c000eea4458c6 (diff) | |
| parent | cccc5a20fce4bde9a4299f8790366790735f1d05 (diff) | |
Merge branch 'master' of https://github.com/yfszzx/stable-diffusion-webui-plus
Diffstat (limited to 'modules/shared.py')
| -rw-r--r-- | modules/shared.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/modules/shared.py b/modules/shared.py index 72513f86..c2ea4186 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -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'])
|