| | |
|---|
| | | text_name="text", |
|---|
| | | non_linguistic_symbols=train_args.non_linguistic_symbols, |
|---|
| | | ) |
|---|
| | | print("start decoding!!!") |
|---|
| | | |
|---|
| | | |
|---|
| | | @torch.no_grad() |
|---|
| | | def __call__(self, text: Union[list, str], cache: list, split_size=20): |
|---|
| | |
|---|
| | | else: |
|---|
| | | precache = "" |
|---|
| | | cache = [] |
|---|
| | | data = {"text": precache + text} |
|---|
| | | data = {"text": precache + " " + text} |
|---|
| | | result = self.preprocessor(data=data, uid="12938712838719") |
|---|
| | | split_text = self.preprocessor.pop_split_text_data(result) |
|---|
| | | mini_sentences = split_to_mini_sentence(split_text, split_size) |
|---|
| | |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | | logging.basicConfig( |
|---|
| | | level=log_level, |
|---|
| | | format="%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s", |
|---|
| | | ) |
|---|
| | | ncpu = kwargs.get("ncpu", 1) |
|---|
| | | torch.set_num_threads(ncpu) |
|---|
| | | |
|---|
| | | if ngpu >= 1 and torch.cuda.is_available(): |
|---|
| | | device = "cuda" |
|---|
