mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-12 11:56:36 +00:00
feat: generalize temporal tiling across video VAEs (#1926)
This commit is contained in:
1 parent
40e605f3f1
commit
6b3edaaf32
11 files changed
+516
-207
No files matched your search
@@ -758,6 +758,15 @@ namespace Hunyuan {
|
||||
return "hunyuan_video_vae";
|
||||
}
|
||||
|
||||
bool supports_temporal_tiling(VAETemporalDirection direction) const override {
|
||||
return direction == VAETemporalDirection::DECODE;
|
||||
}
|
||||
|
||||
int get_temporal_tile_output_scale(VAETemporalDirection direction) const override {
|
||||
SD_UNUSED(direction);
|
||||
return 4;
|
||||
}
|
||||
|
||||
void get_param_tensors(std::map<std::string, ggml_tensor*>& tensors) override {
|
||||
if (!decode_only) {
|
||||
encoder.get_param_tensors(tensors, weight_prefix + ".encoder");
|
||||
|
||||
Reference in new issue
Block a user