diff --git a/vae.hpp b/vae.hpp
index d3793b3..61dc2e9 100644
--- a/vae.hpp
+++ b/vae.hpp
@@ -529,6 +529,7 @@ struct VAE : public GGMLRunner {
                          struct ggml_tensor** output,
                          struct ggml_context* output_ctx)                                                         = 0;
     virtual void get_param_tensors(std::map<std::string, struct ggml_tensor*>& tensors, const std::string prefix) = 0;
+    virtual void enable_conv2d_direct() {};
 };
 
 struct AutoEncoderKL : public VAE {