yindo/rocm-stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/BillyOutlast/rocm-stable-diffusion.cpp.git synced 2026-02-04 03:01:18 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: support applying LoRA at runtime (#969)

Browse Source
This commit is contained in:
leejet
2025-11-13 21:48:44 +08:00
committed by GitHub
parent 59ebdf0bb5
commit 347710f68f
21 changed files with 901 additions and 227 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
name_conversion.cpp
View File

@@ -855,6 +855,49 @@ std::string convert_sep_to_dot(std::string name) {
return name;
}
std::vector<std::string> cond_stage_model_prefix_vec = {
"cond_stage_model.1.",
"cond_stage_model.",
"conditioner.embedders.",
"text_encoders.",
};
std::vector<std::string> diffuison_model_prefix_vec = {
"model.diffusion_model.",
};
std::vector<std::string> first_stage_model_prefix_vec = {
"first_stage_model.",
"vae.",
};
bool is_cond_stage_model_name(const std::string& name) {
for (const auto& prefix : cond_stage_model_prefix_vec) {
if (starts_with(name, prefix) || starts_with(name, "lora." + prefix)) {
return true;
}
}
return false;
}
bool is_diffusion_model_name(const std::string& name) {
for (const auto& prefix : diffuison_model_prefix_vec) {
if (starts_with(name, prefix) || starts_with(name, "lora." + prefix)) {
return true;
}
}
return false;
}
bool is_first_stage_model_name(const std::string& name) {
for (const auto& prefix : first_stage_model_prefix_vec) {
if (starts_with(name, prefix) || starts_with(name, "lora." + prefix)) {
return true;
}
}
return false;
}
std::string convert_tensor_name(std::string name, SDVersion version) {
bool is_lora = false;
bool is_lycoris_underline = false;
@@ -956,9 +999,6 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
// diffusion model
{
std::vector<std::string> diffuison_model_prefix_vec = {
"model.diffusion_model.",
};
for (const auto& prefix : diffuison_model_prefix_vec) {
if (starts_with(name, prefix)) {
name = convert_diffusion_model_name(name.substr(prefix.size()), prefix, version);
@@ -970,12 +1010,6 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
// cond_stage_model
{
std::vector<std::string> cond_stage_model_prefix_vec = {
"cond_stage_model.1.",
"cond_stage_model.",
"conditioner.embedders.",
"text_encoders.",
};
for (const auto& prefix : cond_stage_model_prefix_vec) {
if (starts_with(name, prefix)) {
name = convert_cond_stage_model_name(name.substr(prefix.size()), prefix);
@@ -987,10 +1021,6 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
// first_stage_model
{
std::vector<std::string> first_stage_model_prefix_vec = {
"first_stage_model.",
"vae.",
};
for (const auto& prefix : first_stage_model_prefix_vec) {
if (starts_with(name, prefix)) {
name = convert_first_stage_model_name(name.substr(prefix.size()), prefix);