fix export_jit.py (#2455)

wenet-e2e · Apr 2, 2024 · 1feea35 · 1feea35
1 parent f3a434a
commit 1feea35
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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