DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-01-02 10:43:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add support for more underline loras (#1135)

Browse Source
This commit is contained in:
leejet 2025-12-24 22:59:23 +08:00 committed by GitHub
parent 3e6c428c27
commit 3d5fdd7b37
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
name_conversion.cpp
View File

@ -960,6 +960,7 @@ bool is_first_stage_model_name(const std::string& name) {
std::string convert_tensor_name(std::string name, SDVersion version) { std::string convert_tensor_name(std::string name, SDVersion version) {
bool is_lora = false; bool is_lora = false;
bool is_lycoris_underline = false; bool is_lycoris_underline = false;
bool is_underline = false;
std::vector<std::string> lora_prefix_vec = { std::vector<std::string> lora_prefix_vec = {
"lora.lora.", "lora.lora.",
"lora.lora_", "lora.lora_",
@ -967,12 +968,27 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
"lora.lycoris.", "lora.lycoris.",
"lora.", "lora.",
}; };
std::vector<std::string> underline_lora_prefix_vec = {
"unet_",
"te_",
"te1_",
"te2_",
"te3_",
"vae_",
};
for (const auto& prefix : lora_prefix_vec) { for (const auto& prefix : lora_prefix_vec) {
if (starts_with(name, prefix)) { if (starts_with(name, prefix)) {
is_lora = true; is_lora = true;
name = name.substr(prefix.size()); name = name.substr(prefix.size());
if (contains(prefix, "lycoris_")) { if (contains(prefix, "lycoris_")) {
is_lycoris_underline = true; is_lycoris_underline = true;
} else {
for (const auto& underline_lora_prefix : underline_lora_prefix_vec) {
if (starts_with(name, underline_lora_prefix)) {
is_underline = true;
break;
}
}
} }
break; break;
} }
@ -1034,7 +1050,7 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
// LOG_DEBUG("name %s %d", name.c_str(), version); // LOG_DEBUG("name %s %d", name.c_str(), version);
if (sd_version_is_unet(version) || sd_version_is_flux(version) || 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);
} }
} }