From 1feea35a91ec61ef826cf0ac18ce3013b0dcc321 Mon Sep 17 00:00:00 2001
From: Dinghao Zhou <hamddct@gmail.com>
Date: Tue, 2 Apr 2024 17:27:44 +0800
Subject: [PATCH] fix export_jit.py (#2455)

---
 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