嘉渊 il y a 2 ans
Parent
commit
d6023759a1
1 fichiers modifiés avec 1 ajouts et 1 suppressions
  1. 1 1
      funasr/build_utils/build_diar_model.py

+ 1 - 1
funasr/build_utils/build_diar_model.py

@@ -221,7 +221,7 @@ def build_diar_model(args):
         args.frontend = None
         args.frontend_conf = {}
         frontend = None
-        input_size = None
+        input_size = args.input_size
 
     if args.model == "sond":
         # encoder