use lycoris_lora module

ddPn08 · Jul 7, 2023 · 8c4c847 · 8c4c847
1 parent 994f0b8
commit 8c4c847
Show file tree

Hide file tree

Showing 8 changed files with 189 additions and 889 deletions.
diff --git a/modules/diffusion/networks/__init__.py b/modules/diffusion/networks/__init__.py
@@ -3,7 +3,6 @@
 from glob import glob
 from typing import *
 
-import losalina.hypernetwork
 import torch
 
 from api.events import event_handler
@@ -12,7 +11,7 @@
 from modules.logger import logger
 from modules.shared import ROOT_DIR
 
-from . import lora, lyco
+from . import lyco
 
 latest_networks: List[Tuple[str, torch.nn.Module]] = []
 
@@ -83,13 +82,10 @@ def load_network_modules(e: LoadResourceEvent):
         multiplier = float(multiplier)
         if module_type == "lora":
             filepath = find_network_filepath(basename, "lora")
-            network_module = lora
+            network_module = lyco
         elif module_type == "lyco":
             filepath = find_network_filepath(basename, "lycoris")
             network_module = lyco
-        elif module_type == "hypernet":
-            filepath = find_network_filepath(basename, "hypernetwork")
-            network_module = losalina.hypernetwork
         else:
             continue
 

diff --git a/modules/diffusion/networks/lora.py b/modules/diffusion/networks/lora.py