funasr2

游雁

2023-12-19 00ea1186f96e6732e2edb4fab6c0ed6896e3b352

 funasr/bin/inference.py

@@ -81,7 +81,7 @@

class AutoModel:
   def __init__(self, **kwargs):
      registry_tables.print_register_tables()
      registry_tables.print()
      assert "model" in kwargs
      if "model_conf" not in kwargs:
         logging.info("download models from model hub: {}".format(kwargs.get("model_hub", "ms")))
@@ -108,6 +108,7 @@
         frontend_class = registry_tables.frontend_classes.get(frontend.lower())
         frontend = frontend_class(**kwargs["frontend_conf"])
         kwargs["frontend"] = frontend
         kwargs["input_size"] = frontend.output_size()
      
      # build model
      model_class = registry_tables.model_classes.get(kwargs["model"].lower())