Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-17 16:24:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Check if the model has unet before remove_hidiffusion

Browse Source
This commit is contained in:
Disty0
2024-05-21 18:01:58 +03:00
parent 44857a04a8
commit 585b6037bd
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/hidiffusion/__init__.py
+2 -1
View File
@@ -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)