[fix] fix cli pt model loading

wenet-e2e · Jan 10, 2025 · f84147c · f84147c
1 parent e296cf2
commit f84147c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/wenet/cli/model.py b/wenet/cli/model.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import argparse
 import os
 
 import torch
@@ -204,6 +205,7 @@ def load_model_pt(model_dir):
         configs['cmvn_conf']['cmvn_file'] = cmvn_file
     # Read model
     pt_file = os.path.join(model_dir, 'final.pt')
-    args = {'checkpoint': pt_file}
+    args = argparse.Namespace()
+    args.checkpoint = pt_file
     model, configs = init_model(args, configs)
     return model