From 585b6037bdc7142c4721428a41e4d0a3af8fc4e1 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 21 May 2024 18:01:58 +0300 Subject: [PATCH] Check if the model has unet before remove_hidiffusion --- modules/hidiffusion/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/hidiffusion/__init__.py b/modules/hidiffusion/__init__.py index 6e5e0df3d..649af8ba8 100644 --- a/modules/hidiffusion/__init__.py +++ b/modules/hidiffusion/__init__.py @@ -34,4 +34,5 @@ def apply_hidiffusion(p, model_type): def remove_hidiffusion(p): - hidiffusion.remove_hidiffusion(shared.sd_model) + if hasattr(shared.sd_model, "unet"): + hidiffusion.remove_hidiffusion(shared.sd_model)