From 1e5703fc6e35f0b1e6ba0459c3e8f3cc6a65bee3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?=
 <83450930+Liyulingyue@users.noreply.github.com>
Date: Thu, 5 Dec 2024 23:00:40 +0800
Subject: [PATCH] Update model.py

---
 paddlespeech/s2t/exps/deepspeech2/model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddlespeech/s2t/exps/deepspeech2/model.py b/paddlespeech/s2t/exps/deepspeech2/model.py
index c90e36d45..7836d3adf 100644
--- a/paddlespeech/s2t/exps/deepspeech2/model.py
+++ b/paddlespeech/s2t/exps/deepspeech2/model.py
@@ -641,6 +641,8 @@ class DeepSpeech2ExportTester(DeepSpeech2Tester):
                 self.args.export_path + ".pdmodel",
                 self.args.export_path + ".pdiparams")
 
+        deepspeech_config.disable_mkldnn()
+
         if (os.environ['CUDA_VISIBLE_DEVICES'].strip() != ''):
             deepspeech_config.enable_use_gpu(100, 0)
             deepspeech_config.enable_memory_optim()