mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-11 11:26:34 +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
@@ -819,6 +819,21 @@ struct TinyVideoAutoEncoder : public VAE {
|
||||
return "taehv";
|
||||
}
|
||||
|
||||
bool supports_temporal_tiling(VAETemporalDirection direction) const override {
|
||||
return direction == VAETemporalDirection::DECODE && !sd_version_is_minimax_h3(version);
|
||||
}
|
||||
|
||||
int get_temporal_tile_output_scale(VAETemporalDirection direction) const override {
|
||||
SD_UNUSED(direction);
|
||||
int scale = 1;
|
||||
for (bool upscale : taehv.time_upscale) {
|
||||
if (upscale) {
|
||||
scale *= 2;
|
||||
}
|
||||
}
|
||||
return scale;
|
||||
}
|
||||
|
||||
void get_param_tensors(std::map<std::string, ggml_tensor*>& tensors) override {
|
||||
taehv.get_param_tensors(tensors, weight_prefix);
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user