setup jamo

游雁

2023-11-24 b5d3df75cf6462aa3bf42fd3c86fa2aa7f1c8a15

 funasr/bin/vad_inference_launch.py

@@ -18,7 +18,6 @@

import numpy as np
import torch
from typeguard import check_argument_types
from funasr.build_utils.build_streaming_iterator import build_streaming_iterator
from funasr.fileio.datadir_writer import DatadirWriter
from funasr.torch_utils.set_all_random_seed import set_all_random_seed
@@ -47,7 +46,6 @@
        num_workers: int = 1,
        **kwargs,
):
    assert check_argument_types()
    if batch_size > 1:
        raise NotImplementedError("batch decoding is not implemented")

@@ -92,6 +90,7 @@
            preprocess_args=None,
            data_path_and_name_and_type=data_path_and_name_and_type,
            dtype=dtype,
            fs=fs,
            batch_size=batch_size,
            key_file=key_file,
            num_workers=num_workers,
@@ -125,7 +124,7 @@
                vad_results.append(item)
                if writer is not None:
                    ibest_writer["text"][keys[i]] = "{}".format(results[i])

        torch.cuda.empty_cache()
        return vad_results

    return _forward
@@ -148,7 +147,6 @@
        num_workers: int = 1,
        **kwargs,
):
    assert check_argument_types()

    logging.basicConfig(
        level=log_level,
@@ -192,6 +190,7 @@
            preprocess_args=None,
            data_path_and_name_and_type=data_path_and_name_and_type,
            dtype=dtype,
            fs=fs,
            batch_size=batch_size,
            key_file=key_file,
            num_workers=num_workers,