refactor: optimize the VAE architecture (#1345)

This commit is contained in:
leejet
2026-03-15 16:57:42 +08:00
committed by GitHub
parent 83eabd7c01
commit acc3bf1fdc
8 changed files with 1437 additions and 1359 deletions

View File

@@ -1120,7 +1120,11 @@ std::string convert_tensor_name(std::string name, SDVersion version) {
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);
name = prefix + name;
if (version == VERSION_SDXS) {
name = "tae." + name;
} else {
name = prefix + name;
}
break;
}
}