lycoris-lora 2.1.0.dev7__tar.gz → 2.1.0.dev8__tar.gz

{lycoris_lora-2.1.0.dev7 → lycoris_lora-2.1.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lycoris_lora
-Version: 2.1.0.dev7
+Version: 2.1.0.dev8
 Summary: Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable diffusion
 Home-page: https://github.com/KohakuBlueleaf/LyCORIS
 Author: Shih-Ying Yeh(KohakuBlueLeaf), Yu-Guan Hsieh, Zhidong Gao

{lycoris_lora-2.1.0.dev7 → lycoris_lora-2.1.0.dev8}/lycoris/kohya/__init__.py

@@ -4,13 +4,14 @@
 # https://github.com/cloneofsimo/lora/blob/master/lora_diffusion/lora.py
 # https://github.com/kohya-ss/sd-scripts/blob/main/networks/lora.py
 
-import math
-from warnings import warn
 import os
+import re
 import sys
+from warnings import warn
+from typing import List
 
 sys.setrecursionlimit(10000)
-from typing import List
+
 import torch
 import torch.utils.checkpoint as checkpoint
 
@@ -482,7 +483,9 @@ class LycorisNetworkKohya(LycorisNetwork):
                         ]
                     )
                     next_config = {}
-                elif name in target_replace_names:
+                elif name in target_replace_names or any(
+                    re.match(t, name) for t in target_replace_names
+                ):
                     if name in self.NAME_ALGO_MAP:
                         next_config = self.NAME_ALGO_MAP[name]
                         algo = next_config.get("algo", network_module)

{lycoris_lora-2.1.0.dev7 → lycoris_lora-2.1.0.dev8}/lycoris/utils/__init__.py

@@ -230,12 +230,14 @@ def extract_diff(
                         extract_c.detach().cpu().contiguous().half()
                     )
                     diff = (
-                        root_weight
-                        - torch.einsum(
-                            "i j k l, j r, p i -> p r k l",
-                            extract_c,
-                            extract_a.flatten(1, -1),
-                            extract_b.flatten(1, -1),
+                        (
+                            root_weight
+                            - torch.einsum(
+                                "i j k l, j r, p i -> p r k l",
+                                extract_c,
+                                extract_a.flatten(1, -1),
+                                extract_b.flatten(1, -1),
+                            )
                         )
                         .detach()
                         .cpu()

{lycoris_lora-2.1.0.dev7 → lycoris_lora-2.1.0.dev8}/lycoris_lora.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lycoris-lora
-Version: 2.1.0.dev7
+Version: 2.1.0.dev8
 Summary: Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable diffusion
 Home-page: https://github.com/KohakuBlueleaf/LyCORIS
 Author: Shih-Ying Yeh(KohakuBlueLeaf), Yu-Guan Hsieh, Zhidong Gao

{lycoris_lora-2.1.0.dev7 → lycoris_lora-2.1.0.dev8}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 setup(
     name="lycoris_lora",
     packages=find_packages(),
-    version="2.1.0.dev7",
+    version="2.1.0.dev8",
     url="https://github.com/KohakuBlueleaf/LyCORIS",
     description="Lora beYond Conventional methods, Other Rank adaptation Implementations for Stable diffusion",
     author="Shih-Ying Yeh(KohakuBlueLeaf), Yu-Guan Hsieh, Zhidong Gao",