diff --git a/patches/external_pr/ui.py b/patches/external_pr/ui.py index 1bdaebe..1264d26 100644 --- a/patches/external_pr/ui.py +++ b/patches/external_pr/ui.py @@ -39,7 +39,8 @@ def train_hypernetwork_ui(*args): shared.loaded_hypernetwork = initial_hypernetwork else: shared.loaded_hypernetworks = [] - del hypernetwork + if 'hypernetwork' in locals(): + del hypernetwork gc.collect() shared.sd_model.cond_stage_model.to(devices.device) shared.sd_model.first_stage_model.to(devices.device)