chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Update alignment.py

Browse Source
This commit is contained in:
piuy11
2023-09-25 15:33:06 +09:00
committed by GitHub
parent e94b904308
commit f137f31de6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
whisperx/alignment.py
View File

@ -3,7 +3,7 @@ Forced Alignment with Whisper
C. Max Bain C. Max Bain
""" """
from dataclasses import dataclass from dataclasses import dataclass
from typing import Iterator, Union, List from typing import Iterable, Union, List
import numpy as np import numpy as np
import pandas as pd import pandas as pd
@ -91,7 +91,7 @@ def load_align_model(language_code, device, model_name=None, model_dir=None):
def align( def align(
transcript: Iterator[SingleSegment], transcript: Iterable[SingleSegment],
model: torch.nn.Module, model: torch.nn.Module,
align_model_metadata: dict, align_model_metadata: dict,
audio: Union[str, np.ndarray, torch.Tensor], audio: Union[str, np.ndarray, torch.Tensor],