lycoris-lora 3.0.1.dev2__tar.gz → 3.0.1.dev4__tar.gz

{lycoris_lora-3.0.1.dev2/lycoris_lora.egg-info → lycoris_lora-3.0.1.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lycoris_lora
-Version: 3.0.1.dev2
+Version: 3.0.1.dev4
 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-3.0.1.dev2 → lycoris_lora-3.0.1.dev4}/lycoris/config.py

@@ -7,6 +7,8 @@ PRESET = {
             "Downsample2D",
             "Upsample2D",
             "HunYuanDiTBlock",
+            "DoubleStreamBlock",
+            "SingleStreamBlock",
         ],
         "unet_target_name": [
             "conv_in",
@@ -28,6 +30,8 @@ PRESET = {
             "Transformer2DModel",
             "ResnetBlock2D",
             "HunYuanDiTBlock",
+            "DoubleStreamBlock",
+            "SingleStreamBlock",
         ],
         "unet_target_name": [
             "time_embedding.linear_1",
@@ -46,6 +50,8 @@ PRESET = {
         "unet_target_module": [
             "Transformer2DModel",
             "HunYuanDiTBlock",
+            "DoubleStreamBlock",
+            "SingleStreamBlock",
         ],
         "unet_target_name": [],
         "text_encoder_target_module": [
@@ -78,6 +84,8 @@ PRESET = {
             "Downsample2D",
             "Upsample2D",
             "HunYuanDiTBlock",
+            "DoubleStreamBlock",
+            "SingleStreamBlock",
         ],
         "unet_target_name": [
             "conv_in",
@@ -93,6 +101,8 @@ PRESET = {
         "unet_target_module": [
             "Transformer2DModel",
             "HunYuanDiTBlock",
+            "DoubleStreamBlock",
+            "SingleStreamBlock",
         ],
         "unet_target_name": [],
         "text_encoder_target_module": [],

{lycoris_lora-3.0.1.dev2 → lycoris_lora-3.0.1.dev4}/lycoris/kohya.py

@@ -214,6 +214,8 @@ class LycorisNetworkKohya(LycorisNetwork):
         "Downsample2D",
         "Upsample2D",
         "HunYuanDiTBlock",
+        "DoubleStreamBlock",
+        "SingleStreamBlock",
     ]
     UNET_TARGET_REPLACE_NAME = [
         "conv_in",

{lycoris_lora-3.0.1.dev2 → lycoris_lora-3.0.1.dev4}/lycoris/modules/norms.py

@@ -37,12 +37,14 @@ class NormModule(LycorisBaseModule):
         if self.module_type == "unknown":
             if not hasattr(org_module, "weight") or not hasattr(org_module, "_norm"):
                 warning_once(f"{type(org_module)} is not supported in Norm algo.")
+                self.not_supported=True
                 return
             else:
                 self.dim = org_module.weight.numel()
                 self.not_supported = False
         elif self.module_type not in self.support_module:
             warning_once(f"{self.module_type} is not supported in Norm algo.")
+            self.not_supported=True
             return
 
         self.w_norm = nn.Parameter(torch.zeros(self.dim))

{lycoris_lora-3.0.1.dev2 → lycoris_lora-3.0.1.dev4}/lycoris/wrapper.py

@@ -334,7 +334,7 @@ class LycorisNetwork(torch.nn.Module):
                     next_config = self.MODULE_ALGO_MAP[module_name]
                     next_algo = next_config.get("algo", algo)
                     new_loras, new_lora_names = create_modules_(
-                        f"{prefix}_{name}", module, next_algo, next_config
+                        f"{prefix}_{name}" if name else prefix, module, next_algo, next_config
                     )
                     for lora_name, lora in zip(new_lora_names, new_loras):
                         if lora_name not in loras:

{lycoris_lora-3.0.1.dev2 → lycoris_lora-3.0.1.dev4/lycoris_lora.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lycoris_lora
-Version: 3.0.1.dev2
+Version: 3.0.1.dev4
 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-3.0.1.dev2 → lycoris_lora-3.0.1.dev4}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 setup(
     name="lycoris_lora",
     packages=find_packages(),
-    version="3.0.1.dev2",
+    version="3.0.1.dev4",
     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",