Refactored Metal/mps fixes.

This commit is contained in:
Elias Oenal
2022-09-12 16:32:44 +02:00
parent 5dc05c0d0d
commit b7f95869b4
3 changed files with 27 additions and 38 deletions

View File

@@ -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