From b7f95869b4542d356a12da6860b1e6c227784560 Mon Sep 17 00:00:00 2001 From: Elias Oenal Date: Mon, 12 Sep 2022 16:32:44 +0200 Subject: Refactored Metal/mps fixes. --- modules/shared.py | 2 ++ 1 file changed, 2 insertions(+) (limited to 'modules/shared.py') diff --git a/modules/shared.py b/modules/shared.py index 9eeb64e3..5312768b 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -46,6 +46,8 @@ parser.add_argument("--ui-config-file", type=str, help="filename to use for ui c cmd_opts = parser.parse_args() device = get_optimal_device() +device_codeformer = device +device_seed_type = device batch_cond_uncond = cmd_opts.always_batch_cond_uncond or not (cmd_opts.lowvram or cmd_opts.medvram) parallel_processing_allowed = not cmd_opts.lowvram and not cmd_opts.medvram -- cgit v1.2.3