Add GigaAM NeMo transducer model for Russian ASR (#1467)

This commit is contained in:
Fangjun Kuang
2024-10-25 15:20:13 +08:00
committed by GitHub
parent b41f6d2c94
commit 707cf792c5
12 changed files with 543 additions and 21 deletions

View File

@@ -75,6 +75,7 @@ def add_meta_data(filename: str, meta_data: Dict[str, str]):
onnx.save(model, filename)
@torch.no_grad()
def main():
model = EncDecCTCModel.from_config_file("./ctc_model_config.yaml")
ckpt = torch.load("./ctc_model_weights.ckpt", map_location="cpu")