@@ -3,8 +3,6 @@
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
# MIT License (https://opensource.org/licenses/MIT)
-
import argparse
import logging
import os
@@ -206,3 +206,4 @@ class LMTask(AbsTask):
# 3. Initialize
if args.init is not None:
initialize(model, args.init)
+ return model