[onnx] clone weight for whisper (#2501)

wenet-e2e · Apr 24, 2024 · d9e5ee9 · d9e5ee9
1 parent cb65678
commit d9e5ee9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/wenet/utils/init_model.py b/wenet/utils/init_model.py
@@ -186,6 +186,8 @@ def init_model(args, configs):
 
     # Tie emb.weight to decoder.output_layer.weight
     if model.decoder.tie_word_embedding:
+        if not hasattr(args, 'jit'):
+            args.jit = True  # i.e. export onnx/jit/ipex
         model.decoder.tie_or_clone_weights(jit_mode=args.jit)
 
     return model, configs