diff --git a/modules/sd_models.py b/modules/sd_models.py index ff245b7a6688979a49e4df37b8a6f58a58af8a65..1747ca62103112ae537d99fadea7dd50c8670172 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -659,10 +659,11 @@ def get_empty_cond(sd_model): def send_model_to_cpu(m): - if m.lowvram: - lowvram.send_everything_to_cpu() - else: - m.to(devices.cpu) + if m is not None: + if m.lowvram: + lowvram.send_everything_to_cpu() + else: + m.to(devices.cpu) devices.torch_gc()