From 957d6b3bb429d49b31cc2eac1e90467873bdf941 Mon Sep 17 00:00:00 2001
From: Dinghao Zhou <hamddct@gmail.com>
Date: Tue, 2 Apr 2024 17:26:38 +0800
Subject: [PATCH] fix export_jit.py

---
 wenet/bin/export_jit.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/wenet/bin/export_jit.py b/wenet/bin/export_jit.py
index 70f53586b..98eadd61c 100644
--- a/wenet/bin/export_jit.py
+++ b/wenet/bin/export_jit.py
@@ -38,6 +38,7 @@ def get_args():
 
 def main():
     args = get_args()
+    args.jit = True
     logging.basicConfig(level=logging.DEBUG,
                         format='%(asctime)s %(levelname)s %(message)s')
     # No need gpu for model export