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

This commit is contained in:
leejet authored and GitHub committed 2026-09-19 17:50:01 +08:00
1 parent 2ea8aff7ef
commit 3e037a81e4
9 files changed
+24 -8

No files matched your search

+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) {