diff --git a/patches/external_pr/hypernetwork.py b/patches/external_pr/hypernetwork.py index 2796cc6..25374be 100644 --- a/patches/external_pr/hypernetwork.py +++ b/patches/external_pr/hypernetwork.py @@ -949,4 +949,4 @@ def train_hypernetwork_tuning(id_task, hypernetwork_name, data_root, log_directo move_optimizer, load_hypernetworks_option, load_training_option) if shared.state.interrupted: - return + return None, None diff --git a/patches/external_pr/ui.py b/patches/external_pr/ui.py index 998e690..e43e20d 100644 --- a/patches/external_pr/ui.py +++ b/patches/external_pr/ui.py @@ -48,11 +48,10 @@ def train_hypernetwork_ui_tuning(*args): try: sd_hijack.undo_optimizations() - hypernetwork, filename = train_hypernetwork_tuning(*args) + train_hypernetwork_tuning(*args) res = f""" -Training {'interrupted' if shared.state.interrupted else 'finished'} at {hypernetwork.step} steps. -Hypernetwork saved to {html.escape(filename)} +Training {'interrupted' if shared.state.interrupted else 'finished'}. """ return res, "" except Exception: