1580158
/
Hypernetwork-MonkeyPatch-Extension
mirror of https://github.com/aria1th/Hypernetwork-MonkeyPatch-Extension.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix minor errors

hypernetwork-tuning
aria1th 2023-01-16 06:49:46 +09:00
parent f75fd3bb97
commit 9d2e8bd963
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
patches/external_pr/hypernetwork.py
View File

@ -949,4 +949,4 @@ def train_hypernetwork_tuning(id_task, hypernetwork_name, data_root, log_directo
move_optimizer, move_optimizer,
load_hypernetworks_option, load_training_option) load_hypernetworks_option, load_training_option)
if shared.state.interrupted: if shared.state.interrupted:
return return None, None

5
patches/external_pr/ui.py
View File

@ -48,11 +48,10 @@ def train_hypernetwork_ui_tuning(*args):
try: try:
sd_hijack.undo_optimizations() sd_hijack.undo_optimizations()
hypernetwork, filename = train_hypernetwork_tuning(*args) train_hypernetwork_tuning(*args)
res = f""" res = f"""
Training {'interrupted' if shared.state.interrupted else 'finished'} at {hypernetwork.step} steps. Training {'interrupted' if shared.state.interrupted else 'finished'}.
Hypernetwork saved to {html.escape(filename)}
""" """
return res, "" return res, ""
except Exception: except Exception: