VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typos

Browse Source
This commit is contained in:
Kohya S
2023-07-17 09:05:50 +09:00
parent 39e62b948e
commit 1d4672d747
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -12,7 +12,7 @@ This repository contains the scripts for:
* DreamBooth training, including U-Net and Text Encoder * DreamBooth training, including U-Net and Text Encoder
* Fine-tuning (native training), including U-Net and Text Encoder * Fine-tuning (native training), including U-Net and Text Encoder
* LoRA training * LoRA training
* Texutl Inversion training * Texutal Inversion training
* Image generation * Image generation
* Model conversion (supports 1.x and 2.x, Stable Diffision ckpt/safetensors and Diffusers) * Model conversion (supports 1.x and 2.x, Stable Diffision ckpt/safetensors and Diffusers)

2
networks/lora_diffusers.py
View File

@@ -431,7 +431,7 @@ class LoRANetwork(torch.nn.Module):
print(f"weights are restored") print(f"weights are restored")
def load_state_dict(self, state_dict: Mapping[str, Any], strict: bool = True): def load_state_dict(self, state_dict: Mapping[str, Any], strict: bool = True):
# conver SDXL Stability AI's state dict to Diffusers' based state dict # convert SDXL Stability AI's state dict to Diffusers' based state dict
map_keys = list(UNET_CONVERSION_MAP.keys()) # prefix of U-Net modules map_keys = list(UNET_CONVERSION_MAP.keys()) # prefix of U-Net modules
map_keys.sort() map_keys.sort()
for key in list(state_dict.keys()): for key in list(state_dict.keys()):