We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e411520 commit 3959109Copy full SHA for 3959109
name_conversion.cpp
@@ -842,6 +842,7 @@ std::string convert_sep_to_dot(std::string name) {
842
"conv_in",
843
"conv_out",
844
"lora_down",
845
+ "lora_mid",
846
"lora_up",
847
"diff_b",
848
"hada_w1_a",
@@ -997,10 +998,13 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
997
998
if (is_lora) {
999
std::map<std::string, std::string> lora_suffix_map = {
1000
{".lora_down.weight", ".weight.lora_down"},
1001
+ {".lora_mid.weight", ".weight.lora_mid"},
1002
{".lora_up.weight", ".weight.lora_up"},
1003
{".lora.down.weight", ".weight.lora_down"},
1004
+ {".lora.mid.weight", ".weight.lora_mid"},
1005
{".lora.up.weight", ".weight.lora_up"},
1006
{"_lora.down.weight", ".weight.lora_down"},
1007
+ {"_lora.mid.weight", ".weight.lora_mid"},
1008
{"_lora.up.weight", ".weight.lora_up"},
1009
{".lora_A.weight", ".weight.lora_down"},
1010
{".lora_B.weight", ".weight.lora_up"},
0 commit comments