From 39baf0b41bd17b19c532cd1498433fb1649cd1b0 Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Thu, 28 Sep 2023 15:07:23 +0200
Subject: [PATCH] [`PEFT` / `LoRA` ] Kohya checkpoints support (#5202)

kohya support
---
 src/diffusers/utils/state_dict_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/diffusers/utils/state_dict_utils.py b/src/diffusers/utils/state_dict_utils.py
index acc64a5034..1164dba180 100644
--- a/src/diffusers/utils/state_dict_utils.py
+++ b/src/diffusers/utils/state_dict_utils.py
@@ -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 = {