lycoris-lora 2.3.0.dev5__tar.gz → 2.3.0.dev6__tar.gz

{lycoris_lora-2.3.0.dev5/lycoris_lora.egg-info → lycoris_lora-2.3.0.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lycoris_lora
-Version: 2.3.0.dev5
+Version: 2.3.0.dev6
 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.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/kohya/__init__.py

@@ -146,7 +146,7 @@ def create_network(
             weight_decompose=weight_decompose,
             full_matrix=full_matrix,
             bypass_mode=bypass_mode,
-            rs_lora=rs_lora
+            rs_lora=rs_lora,
             unbalanced_factorization=unbalanced_factorization,
         )
 

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/kohya/model_utils.py

@@ -357,12 +357,12 @@ def convert_ldm_unet_checkpoint(v2, checkpoint, config):
         attentions = [key for key in input_blocks[i] if f"input_blocks.{i}.1" in key]
 
         if f"input_blocks.{i}.0.op.weight" in unet_state_dict:
-            new_checkpoint[f"down_blocks.{block_id}.downsamplers.0.conv.weight"] = (
-                unet_state_dict.pop(f"input_blocks.{i}.0.op.weight")
-            )
-            new_checkpoint[f"down_blocks.{block_id}.downsamplers.0.conv.bias"] = (
-                unet_state_dict.pop(f"input_blocks.{i}.0.op.bias")
-            )
+            new_checkpoint[
+                f"down_blocks.{block_id}.downsamplers.0.conv.weight"
+            ] = unet_state_dict.pop(f"input_blocks.{i}.0.op.weight")
+            new_checkpoint[
+                f"down_blocks.{block_id}.downsamplers.0.conv.bias"
+            ] = unet_state_dict.pop(f"input_blocks.{i}.0.op.bias")
 
         paths = renew_resnet_paths(resnets)
         meta_path = {
@@ -457,12 +457,12 @@ def convert_ldm_unet_checkpoint(v2, checkpoint, config):
                 index = list(output_block_list.values()).index(
                     ["conv.bias", "conv.weight"]
                 )
-                new_checkpoint[f"up_blocks.{block_id}.upsamplers.0.conv.bias"] = (
-                    unet_state_dict[f"output_blocks.{i}.{index}.conv.bias"]
-                )
-                new_checkpoint[f"up_blocks.{block_id}.upsamplers.0.conv.weight"] = (
-                    unet_state_dict[f"output_blocks.{i}.{index}.conv.weight"]
-                )
+                new_checkpoint[
+                    f"up_blocks.{block_id}.upsamplers.0.conv.bias"
+                ] = unet_state_dict[f"output_blocks.{i}.{index}.conv.bias"]
+                new_checkpoint[
+                    f"up_blocks.{block_id}.upsamplers.0.conv.weight"
+                ] = unet_state_dict[f"output_blocks.{i}.{index}.conv.weight"]
 
                 # Clear attentions as they have been attributed above.
                 if len(attentions) == 2:
@@ -586,12 +586,12 @@ def convert_ldm_vae_checkpoint(checkpoint, config):
         ]
 
         if f"encoder.down.{i}.downsample.conv.weight" in vae_state_dict:
-            new_checkpoint[f"encoder.down_blocks.{i}.downsamplers.0.conv.weight"] = (
-                vae_state_dict.pop(f"encoder.down.{i}.downsample.conv.weight")
-            )
-            new_checkpoint[f"encoder.down_blocks.{i}.downsamplers.0.conv.bias"] = (
-                vae_state_dict.pop(f"encoder.down.{i}.downsample.conv.bias")
-            )
+            new_checkpoint[
+                f"encoder.down_blocks.{i}.downsamplers.0.conv.weight"
+            ] = vae_state_dict.pop(f"encoder.down.{i}.downsample.conv.weight")
+            new_checkpoint[
+                f"encoder.down_blocks.{i}.downsamplers.0.conv.bias"
+            ] = vae_state_dict.pop(f"encoder.down.{i}.downsample.conv.bias")
 
         paths = renew_vae_resnet_paths(resnets)
         meta_path = {"old": f"down.{i}.block", "new": f"down_blocks.{i}.resnets"}
@@ -639,12 +639,12 @@ def convert_ldm_vae_checkpoint(checkpoint, config):
         ]
 
         if f"decoder.up.{block_id}.upsample.conv.weight" in vae_state_dict:
-            new_checkpoint[f"decoder.up_blocks.{i}.upsamplers.0.conv.weight"] = (
-                vae_state_dict[f"decoder.up.{block_id}.upsample.conv.weight"]
-            )
-            new_checkpoint[f"decoder.up_blocks.{i}.upsamplers.0.conv.bias"] = (
-                vae_state_dict[f"decoder.up.{block_id}.upsample.conv.bias"]
-            )
+            new_checkpoint[
+                f"decoder.up_blocks.{i}.upsamplers.0.conv.weight"
+            ] = vae_state_dict[f"decoder.up.{block_id}.upsample.conv.weight"]
+            new_checkpoint[
+                f"decoder.up_blocks.{i}.upsamplers.0.conv.bias"
+            ] = vae_state_dict[f"decoder.up.{block_id}.upsample.conv.bias"]
 
         paths = renew_vae_resnet_paths(resnets)
         meta_path = {"old": f"up.{block_id}.block", "new": f"up_blocks.{i}.resnets"}

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/kohya/sdxl_model_util.py

@@ -15,9 +15,7 @@ VAE_SCALE_FACTOR = 0.13025
 MODEL_VERSION_SDXL_BASE_V0_9 = "sdxl_base_v0-9"
 
 # Diffusersの設定を読み込むための参照モデル
-DIFFUSERS_REF_MODEL_ID_SDXL = (
-    "stabilityai/stable-diffusion-xl-base-0.9"  # アクセス権が必要
-)
+DIFFUSERS_REF_MODEL_ID_SDXL = "stabilityai/stable-diffusion-xl-base-0.9"  # アクセス権が必要
 
 DIFFUSERS_SDXL_UNET_CONFIG = {
     "act_fn": "silu",
@@ -237,9 +235,9 @@ def load_models_from_sdxl_checkpoint(model_version, ckpt_path, map_location):
     te2_sd = {}
     for k in list(state_dict.keys()):
         if k.startswith("conditioner.embedders.0.transformer."):
-            te1_sd[k.replace("conditioner.embedders.0.transformer.", "")] = (
-                state_dict.pop(k)
-            )
+            te1_sd[
+                k.replace("conditioner.embedders.0.transformer.", "")
+            ] = state_dict.pop(k)
         elif k.startswith("conditioner.embedders.1.model."):
             te2_sd[k] = state_dict.pop(k)
 

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/modules/glokr.py

@@ -195,7 +195,7 @@ class LokrModule(ModuleCustomSD):
         if self.use_w2 and self.use_w1:
             # use scale = 1
             alpha = lora_dim
-        
+
         r_factor = lora_dim
         if self.rs_lora:
             r_factor = math.sqrt(r_factor)

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/modules/glora.py

@@ -69,7 +69,7 @@ class GLoRAModule(ModuleCustomSD):
             r_factor = math.sqrt(r_factor)
 
         self.scale = alpha / r_factor
-    
+
         self.register_buffer("alpha", torch.tensor(alpha))  # 定数として扱える
 
         # same as microsoft's

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/modules/lilora.py

@@ -46,7 +46,7 @@ class LiLoConModule(ModuleCustomSD):
         self.lora_dim = lora_dim
         self.tucker = False
         self.rs_lora = rs_lora
-        
+
         assert not (
             bypass_mode and weight_decompose
         ), "bypass_mode and dora_wd cannot be used together"

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/modules/locon.py

@@ -141,7 +141,7 @@ class LoConModule(ModuleCustomSD):
             r_factor = math.sqrt(r_factor)
 
         self.scale = alpha / r_factor
-    
+
         self.register_buffer("alpha", torch.tensor(alpha))  # 定数として扱える
 
         if use_scalar:

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6}/lycoris/modules/loha.py

@@ -200,7 +200,7 @@ class LohaModule(ModuleCustomSD):
         if type(alpha) == torch.Tensor:
             alpha = alpha.detach().float().numpy()  # without casting, bf16 causes error
         alpha = lora_dim if alpha is None or alpha == 0 else alpha
-        
+
         r_factor = lora_dim
         if self.rs_lora:
             r_factor = math.sqrt(r_factor)

{lycoris_lora-2.3.0.dev5 → lycoris_lora-2.3.0.dev6/lycoris_lora.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lycoris_lora
-Version: 2.3.0.dev5
+Version: 2.3.0.dev6
 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.3.0.dev5 → lycoris_lora-2.3.0.dev6}/setup.py

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