From 16e3e4023f9b84ab42faccb6b13bc5b9ede4bdcf Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Fri, 15 Mar 2024 18:31:19 -0400 Subject: [PATCH] remove device reference from offload call --- 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 ef37edf56..91d793f82 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -683,7 +683,7 @@ def set_diffuser_options(sd_model, vae = None, op: str = 'model'): shared.opts.diffusers_move_refiner = False shared.log.warning(f'Disabling {op} "Move model to CPU" since "Model CPU offload" is enabled') if not hasattr(sd_model, "_all_hooks") or len(sd_model._all_hooks) == 0: # pylint: disable=protected-access - sd_model.enable_model_cpu_offload(device=devices.device) + sd_model.enable_model_cpu_offload() else: sd_model.maybe_free_model_hooks() sd_model.has_accelerate = True @@ -695,7 +695,7 @@ def set_diffuser_options(sd_model, vae = None, op: str = 'model'): shared.opts.diffusers_move_unet = False shared.opts.diffusers_move_refiner = False shared.log.warning(f'Disabling {op} "Move model to CPU" since "Sequential CPU offload" is enabled') - sd_model.enable_sequential_cpu_offload(device=devices.device) + sd_model.enable_sequential_cpu_offload() sd_model.has_accelerate = True if hasattr(sd_model, "enable_vae_slicing"): if shared.cmd_opts.lowvram or shared.opts.diffusers_vae_slicing: