perf: accelerate VAE direct 3D convolutions (#1996)

This commit is contained in:
leejet
2026-09-19 17:50:01 +08:00
committed by GitHub
parent 2ea8aff7ef
commit 3e037a81e4
9 changed files with 24 additions and 8 deletions
+3 -1
View File
@@ -584,10 +584,12 @@ namespace sd::model_builders {
}
if (sd_ctx_params->vae_conv_direct) {
LOG_INFO("Using Conv2d direct in the vae model");
LOG_INFO("Using Conv2d/Conv3d direct in the vae model");
result.vae->set_conv2d_direct_enabled(true);
result.vae->set_conv3d_direct_enabled(true);
if (result.preview) {
result.preview->set_conv2d_direct_enabled(true);
result.preview->set_conv3d_direct_enabled(true);
}
}
if (result.vae) {