AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[PEFT / LoRA ] Kohya checkpoints support (#5202)

Browse Source 
kohya support
This commit is contained in:
Younes Belkada
2023-09-28 15:07:23 +02:00
committed by GitHub
parent 1c4c4c48d9
commit 39baf0b41b
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/utils/state_dict_utils.py
View File

@@ -37,6 +37,8 @@ DIFFUSERS_TO_PEFT = {
".v_proj.lora_linear_layer.down": ".v_proj.lora_A",
".out_proj.lora_linear_layer.up": ".out_proj.lora_B",
".out_proj.lora_linear_layer.down": ".out_proj.lora_A",
".lora_linear_layer.up": ".lora_B",
".lora_linear_layer.down": ".lora_A",
}
DIFFUSERS_OLD_TO_PEFT = {