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

feat: support 16 channel tae (taesd/taef1) (#527)

Browse Source
This commit is contained in:
stduhpf
2024-12-28 06:13:48 +01:00
committed by GitHub
parent b5cc1422da
commit d50473dc49
2 changed files with 17 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
stable-diffusion.cpp
View File

@@ -360,7 +360,7 @@ public:
first_stage_model->alloc_params_buffer();
first_stage_model->get_param_tensors(tensors, "first_stage_model");
} else {
tae_first_stage = std::make_shared<TinyAutoEncoder>(backend, model_loader.tensor_storages_types, "decoder.layers", vae_decode_only);
tae_first_stage = std::make_shared<TinyAutoEncoder>(backend, model_loader.tensor_storages_types, "decoder.layers", vae_decode_only, version);
}
// first_stage_model->get_param_tensors(tensors, "first_stage_model.");