refactor: unify model source and weight lifecycle management (#1956)

This commit is contained in:
leejet authored and GitHub committed 2026-09-10 23:53:48 +08:00
1 parent d04e8950c1
commit 6b47fec013
36 files changed
+2548 -1462

No files matched your search

+4 -3
View File
@@ -1476,8 +1476,8 @@ struct LTXVideoVAE : public VAE {
ggml_backend_t backend = sd_backend_cpu_init();
LOG_INFO("loading ltx vae from '%s'", model_path.c_str());
auto model_manager = std::make_shared<ModelManager>();
ModelLoader& model_loader = model_manager->loader();
auto model_manager = std::make_shared<ModelManager>();
ModelLoader model_loader;
if (!model_loader.init_from_file_and_convert_name(model_path, "vae.")) {
LOG_ERROR("init model loader from file failed: '%s'", model_path.c_str());
return;
@@ -1491,7 +1491,8 @@ struct LTXVideoVAE : public VAE {
VERSION_LTXAV,
model_manager);
if (!model_manager->register_runner_params("LTX VAE test",
if (!model_manager->set_loader(model_loader) ||
!model_manager->register_runner_params(ModelComponent::VAE,
*vae,
ModelManager::ResidencyMode::ParamBackend,
backend,