fix: preserve Krea2 text-fusion names in Kohya LoRAs (#2301)

This commit is contained in:
leejet authored and GitHub committed 2026-11-09 02:11:28 +08:00
1 parent cd5594e9f9
commit b43ce4badd
1 file changed
+8 -3
+8 -3
View File
@@ -1403,8 +1403,8 @@ std::string convert_pmid_v2_name(const std::string& name) {
return name; return name;
} }
std::string convert_sep_to_dot(std::string name) { std::string convert_sep_to_dot(std::string name, SDVersion version) {
const std::vector<std::string> protected_tokens = { std::vector<std::string> protected_tokens = {
"self_attn", "self_attn",
"out_proj", "out_proj",
"q_proj", "q_proj",
@@ -1480,6 +1480,11 @@ std::string convert_sep_to_dot(std::string name) {
"output_proj", "output_proj",
"token_refiner", "token_refiner",
}; };
if (sd_version_is_krea2(version)) {
protected_tokens.push_back("text_fusion");
protected_tokens.push_back("layerwise_blocks");
protected_tokens.push_back("refiner_blocks");
}
// record the positions of underscores that should NOT be replaced // record the positions of underscores that should NOT be replaced
std::unordered_set<size_t> protected_positions; std::unordered_set<size_t> protected_positions;
@@ -1729,7 +1734,7 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
// LOG_VERBOSE("name %s %d", name.c_str(), version); // LOG_VERBOSE("name %s %d", name.c_str(), version);
if (sd_version_is_unet(version) || is_underline || is_lycoris_underline) { if (sd_version_is_unet(version) || is_underline || is_lycoris_underline) {
name = convert_sep_to_dot(name); name = convert_sep_to_dot(name, version);
} }
} }