| | |
|---|
| | | seaco_decoder = kwargs.get("seaco_decoder", None) |
|---|
| | | if seaco_decoder is not None: |
|---|
| | | seaco_decoder_conf = kwargs.get("seaco_decoder_conf") |
|---|
| | | seaco_decoder_class = tables.decoder_classes.get(seaco_decoder.lower()) |
|---|
| | | seaco_decoder_class = tables.decoder_classes.get(seaco_decoder) |
|---|
| | | self.seaco_decoder = seaco_decoder_class( |
|---|
| | | vocab_size=self.vocab_size, |
|---|
| | | encoder_output_size=self.inner_dim, |
|---|
| | |
|---|
| | | meta_data[ |
|---|
| | | "batch_data_time"] = speech_lengths.sum().item() * frontend.frame_shift * frontend.lfr_n / 1000 |
|---|
| | | |
|---|
| | | speech.to(device=kwargs["device"]), speech_lengths.to(device=kwargs["device"]) |
|---|
| | | speech = speech.to(device=kwargs["device"]) |
|---|
| | | speech_lengths = speech_lengths.to(device=kwargs["device"]) |
|---|
| | | |
|---|
| | | # hotword |
|---|
| | | self.hotword_list = self.generate_hotwords_list(kwargs.get("hotword", None), tokenizer=tokenizer, frontend=frontend) |
|---|
