Skip to content

Commit 3959109

Browse files
authored
fix: improve LoCon support with other naming conventions (#1239)
1 parent e411520 commit 3959109

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

name_conversion.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -842,6 +842,7 @@ std::string convert_sep_to_dot(std::string name) {
842842
"conv_in",
843843
"conv_out",
844844
"lora_down",
845+
"lora_mid",
845846
"lora_up",
846847
"diff_b",
847848
"hada_w1_a",
@@ -997,10 +998,13 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
997998
if (is_lora) {
998999
std::map<std::string, std::string> lora_suffix_map = {
9991000
{".lora_down.weight", ".weight.lora_down"},
1001+
{".lora_mid.weight", ".weight.lora_mid"},
10001002
{".lora_up.weight", ".weight.lora_up"},
10011003
{".lora.down.weight", ".weight.lora_down"},
1004+
{".lora.mid.weight", ".weight.lora_mid"},
10021005
{".lora.up.weight", ".weight.lora_up"},
10031006
{"_lora.down.weight", ".weight.lora_down"},
1007+
{"_lora.mid.weight", ".weight.lora_mid"},
10041008
{"_lora.up.weight", ".weight.lora_up"},
10051009
{".lora_A.weight", ".weight.lora_down"},
10061010
{".lora_B.weight", ".weight.lora_up"},

0 commit comments

Comments
 (0)