Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/apex.git" did not exist on "3fe352118d2fe53dab734e0995a730ca5ffc9cc4"
Unverified Commit 39baf0b4 authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

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

kohya support
parent 1c4c4c48
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/utils/state_dict_utils.py
View file @ 39baf0b4
...@@ -37,6 +37,8 @@ DIFFUSERS_TO_PEFT = { ...@@ -37,6 +37,8 @@ DIFFUSERS_TO_PEFT = {
".v_proj.lora_linear_layer.down": ".v_proj.lora_A", ".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.up": ".out_proj.lora_B",
".out_proj.lora_linear_layer.down": ".out_proj.lora_A", ".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 = { DIFFUSERS_OLD_TO_PEFT = {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment