DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-03-24 02:08:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: correct sdapi LoRA file handling (#1276)

Browse Source
This commit is contained in:
Wagner Bruna 2026-03-01 10:57:06 -03:00 committed by GitHub
parent e64baa3611
commit 60889bc9a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
examples/server/main.cpp
View File

@ -266,6 +266,7 @@ void sd_log_cb(enum sd_log_level_t level, const char* log, void* data) {
struct LoraEntry { struct LoraEntry {
std::string name; std::string name;
std::string path; std::string path;
std::string fullpath;
}; };
int main(int argc, const char** argv) { int main(int argc, const char** argv) {
@ -321,7 +322,8 @@ int main(int argc, const char** argv) {
LoraEntry e; LoraEntry e;
e.name = p.stem().u8string(); e.name = p.stem().u8string();
std::string rel = fs::relative(p, lora_dir).u8string(); e.fullpath = p.u8string();
std::string rel = p.lexically_relative(lora_dir).u8string();
std::replace(rel.begin(), rel.end(), '\\', '/'); std::replace(rel.begin(), rel.end(), '\\', '/');
e.path = rel; e.path = rel;
@ -340,10 +342,11 @@ int main(int argc, const char** argv) {
} }
}; };
auto is_valid_lora_path = [&](const std::string& path) -> bool { auto get_lora_full_path = [&](const std::string& path) -> std::string {
std::lock_guard<std::mutex> lock(lora_mutex); std::lock_guard<std::mutex> lock(lora_mutex);
return std::any_of(lora_cache.begin(), lora_cache.end(), auto it = std::find_if(lora_cache.begin(), lora_cache.end(),
[&](const LoraEntry& e) { return e.path == path; }); [&](const LoraEntry& e) { return e.path == path; });
return (it != lora_cache.end()) ? it->fullpath : "";
}; };
httplib::Server svr; httplib::Server svr;
@ -862,11 +865,12 @@ int main(int argc, const char** argv) {
return bad("lora.path required"); return bad("lora.path required");
} }
if (!is_valid_lora_path(path)) { std::string fullpath = get_lora_full_path(path);
if (fullpath.empty()) {
return bad("invalid lora path: " + path); return bad("invalid lora path: " + path);
} }
lora_path_storage.push_back(path); lora_path_storage.push_back(fullpath);
sd_lora_t l; sd_lora_t l;
l.is_high_noise = is_high_noise; l.is_high_noise = is_high_noise;
l.multiplier = multiplier; l.multiplier = multiplier;