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

Merge pull request #487 from piuy11/main

Browse Source 
Update alignment.py
This commit is contained in:
Max Bain
2023-09-26 14:17:46 -07:00
committed by GitHub
parent 18abcf46ee f137f31de6
commit 2a11ce3ef0
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
"""
from dataclasses import dataclass
from typing import Iterator, Union, List
from typing import Iterable, Union, List
import numpy as np
import pandas as pd
@ -91,7 +91,7 @@ def load_align_model(language_code, device, model_name=None, model_dir=None):
def align(
transcript: Iterator[SingleSegment],
transcript: Iterable[SingleSegment],
model: torch.nn.Module,
align_model_metadata: dict,
audio: Union[str, np.ndarray, torch.Tensor],