more renames

2 files changed, 4 insertions, 4 deletions

diff --git a/modules/hypernetworks/ui.py b/modules/hypernetworks/ui.py
index 811bc31e..e7540f41 100644
--- a/modules/hypernetworks/ui.py
+++ b/modules/hypernetworks/ui.py
@@ -13,7 +13,7 @@ def create_hypernetwork(name):
     fn = os.path.join(shared.cmd_opts.hypernetwork_dir, f"{name}.pt")

     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)

 

     shared.reload_hypernetworks()

@@ -28,7 +28,7 @@ def train_hypernetwork(*args):
     try:

         sd_hijack.undo_optimizations()

 

-        hypernetwork, filename = modules.hypernetwork.hypernetwork.train_hypernetwork(*args)

+        hypernetwork, filename = modules.hypernetworks.hypernetwork.train_hypernetwork(*args)

 

         res = f"""

 Training {'interrupted' if shared.state.interrupted else 'finished'} at {hypernetwork.step} steps.

diff --git a/modules/ui.py b/modules/ui.py
index 42e5d866..ee333c3b 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1111,7 +1111,7 @@ def create_ui(wrap_gradio_gpu_call):
         )

 

         create_hypernetwork.click(

-            fn=modules.hypernetwork.ui.create_hypernetwork,

+            fn=modules.hypernetworks.ui.create_hypernetwork,

             inputs=[

                 new_hypernetwork_name,

             ],

@@ -1164,7 +1164,7 @@ def create_ui(wrap_gradio_gpu_call):
         )

 

         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",

             inputs=[

                 train_hypernetwork_name,