From c8af2affafd0740a7330925dc1966ac9e0fb169b Mon Sep 17 00:00:00 2001 From: Disty0 Date: Fri, 28 Jul 2023 21:26:54 +0300 Subject: [PATCH] Fix original backend reloading --- modules/sd_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/sd_models.py b/modules/sd_models.py index 8673af7ff..b340902b6 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -935,7 +935,7 @@ def reload_model_weights(sd_model=None, info=None, reuse_dict=False, op='model') return if shared.cmd_opts.lowvram or shared.cmd_opts.medvram: lowvram.send_everything_to_cpu() - elif not shared.opts.diffusers_seq_cpu_offload: + elif shared.backend == shared.Backend.ORIGINAL or not shared.opts.diffusers_seq_cpu_offload: sd_model.to(devices.cpu) if reuse_dict or (shared.opts.model_reuse_dict and sd_model is not None): shared.log.info('Reusing previous model dictionary') @@ -979,7 +979,7 @@ def unload_model_weights(op='model'): from modules import sd_hijack if op == 'model' or op == 'dict': if model_data.sd_model: - if not shared.opts.diffusers_seq_cpu_offload: + if shared.backend == shared.Backend.ORIGINAL or not shared.opts.diffusers_seq_cpu_offload: model_data.sd_model.to(devices.cpu) if shared.backend == shared.Backend.ORIGINAL: sd_hijack.model_hijack.undo_hijack(model_data.sd_model)