|
|
|
@ -13,7 +13,7 @@ def create_hypernetwork(name):
|
|
|
|
fn = os.path.join(shared.cmd_opts.hypernetwork_dir, f"{name}.pt")
|
|
|
|
fn = os.path.join(shared.cmd_opts.hypernetwork_dir, f"{name}.pt")
|
|
|
|
assert not os.path.exists(fn), f"file {fn} already exists"
|
|
|
|
assert not os.path.exists(fn), f"file {fn} already exists"
|
|
|
|
|
|
|
|
|
|
|
|
hypernet = modules.hypernetwork.hypernetwork.Hypernetwork(name=name)
|
|
|
|
hypernet = modules.hypernetworks.hypernetwork.Hypernetwork(name=name)
|
|
|
|
hypernet.save(fn)
|
|
|
|
hypernet.save(fn)
|
|
|
|
|
|
|
|
|
|
|
|
shared.reload_hypernetworks()
|
|
|
|
shared.reload_hypernetworks()
|
|
|
|
@ -28,7 +28,7 @@ def train_hypernetwork(*args):
|
|
|
|
try:
|
|
|
|
try:
|
|
|
|
sd_hijack.undo_optimizations()
|
|
|
|
sd_hijack.undo_optimizations()
|
|
|
|
|
|
|
|
|
|
|
|
hypernetwork, filename = modules.hypernetwork.hypernetwork.train_hypernetwork(*args)
|
|
|
|
hypernetwork, filename = modules.hypernetworks.hypernetwork.train_hypernetwork(*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'} at {hypernetwork.step} steps.
|
|
|
|
|