atsr

语帆

2024-03-01 21b49bd56f44ea5921bacd8c0a3c5e35680cb405

atsr

 funasr/auto/auto_model.py

@@ -213,7 +213,7 @@
        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 @@
    if kwargs.get("debug", False):
        import pdb; pdb.set_trace()
    model = AutoModel(**kwargs)
    pdb.set_trace()
    res = model.generate(input=kwargs["input"])
    print(res)