atsr

funasr/auto/auto_model.py

@@ -213,7 +213,7 @@ class AutoModel:
         batch_size = kwargs.get("batch_size", 1)
         # if kwargs.get("device", "cpu") == "cpu":
         #     batch_size = 1
-        pdb.set_trace()
+
         key_list, data_list = prepare_data_iterator(input, input_len=input_len, data_type=kwargs.get("data_type", None), key=key)
 
         speed_stats = {}

funasr/bin/inference.py

@@ -1,7 +1,7 @@
 import hydra
 import logging
 from omegaconf import DictConfig, OmegaConf, ListConfig
-import pdb
+
 from funasr.auto.auto_model import AutoModel
 
 
@@ -23,7 +23,6 @@ def main_hydra(cfg: DictConfig):
     if kwargs.get("debug", False):
         import pdb; pdb.set_trace()
     model = AutoModel(**kwargs)
-    pdb.set_trace()
     res = model.generate(input=kwargs["input"])
     print(res)