modularize ui imports

This commit is contained in:
Vladimir Mandic
2023-12-22 08:07:34 -05:00
parent ec0a08c4db
commit 4b74d3ebf9
4 changed files with 127 additions and 115 deletions
+2 -2
View File
@@ -23,8 +23,8 @@ Training {'interrupted' if shared.state.interrupted else 'finished'} at {hyperne
Hypernetwork saved to {html.escape(filename)}
"""
return res, ""
except Exception:
raise
except Exception as e:
raise RuntimeError("Hypernetwork error") from e
finally:
shared.sd_model.cond_stage_model.to(devices.device)
shared.sd_model.first_stage_model.to(devices.device)