Commit b0583be0 authored by AUTOMATIC's avatar AUTOMATIC

more renames

parent 873efeed
...@@ -13,7 +13,7 @@ def create_hypernetwork(name): ...@@ -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): ...@@ -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.
......
...@@ -1111,7 +1111,7 @@ def create_ui(wrap_gradio_gpu_call): ...@@ -1111,7 +1111,7 @@ def create_ui(wrap_gradio_gpu_call):
) )
create_hypernetwork.click( create_hypernetwork.click(
fn=modules.hypernetwork.ui.create_hypernetwork, fn=modules.hypernetworks.ui.create_hypernetwork,
inputs=[ inputs=[
new_hypernetwork_name, new_hypernetwork_name,
], ],
...@@ -1164,7 +1164,7 @@ def create_ui(wrap_gradio_gpu_call): ...@@ -1164,7 +1164,7 @@ def create_ui(wrap_gradio_gpu_call):
) )
train_hypernetwork.click( train_hypernetwork.click(
fn=wrap_gradio_gpu_call(modules.hypernetwork.ui.train_hypernetwork, extra_outputs=[gr.update()]), fn=wrap_gradio_gpu_call(modules.hypernetworks.ui.train_hypernetwork, extra_outputs=[gr.update()]),
_js="start_training_textual_inversion", _js="start_training_textual_inversion",
inputs=[ inputs=[
train_hypernetwork_name, train_hypernetwork_name,
......
...@@ -83,7 +83,7 @@ modules.scripts.load_scripts(os.path.join(script_path, "scripts")) ...@@ -83,7 +83,7 @@ modules.scripts.load_scripts(os.path.join(script_path, "scripts"))
shared.sd_model = modules.sd_models.load_model() shared.sd_model = modules.sd_models.load_model()
shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(shared.sd_model))) shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(shared.sd_model)))
shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetwork.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork))) shared.opts.onchange("sd_hypernetwork", wrap_queued_call(lambda: modules.hypernetworks.hypernetwork.load_hypernetwork(shared.opts.sd_hypernetwork)))
def webui(): def webui():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment